A Recursive Algebraic Coloring Technique for Hardware-Efficient Symmetric Sparse Matrix-Vector Multiplication

TL;DR

This paper introduces RACE, a recursive algebraic coloring algorithm that significantly improves hardware efficiency and scalability for symmetric sparse matrix-vector multiplication on multicore systems.

Contribution

The paper presents RACE, a novel coloring algorithm and open-source library that overcomes limitations of existing methods in load balancing and memory hierarchy utilization.

Findings

RACE outperforms state-of-the-art coloring techniques and Intel MKL on multiple matrices.

RACE scales efficiently on multicore processors, aligning with the Roofline model.

The approach is applicable to various sparse matrix operations with data dependencies.

Abstract

The symmetric sparse matrix-vector multiplication (SymmSpMV) is an important building block for many numerical linear algebra kernel operations or graph traversal applications. Parallelizing SymmSpMV on today's multicore platforms with up to 100 cores is difficult due to the need to manage conflicting updates on the result vector. Coloring approaches can be used to solve this problem without data duplication, but existing coloring algorithms do not take load balancing and deep memory hierarchies into account, hampering scalability and full-chip performance. In this work, we propose the recursive algebraic coloring engine (RACE), a novel coloring algorithm and open-source library implementation, which eliminates the shortcomings of previous coloring methods in terms of hardware efficiency and parallelization overhead. We describe the level construction, distance-k coloring, and load balancing steps in RACE, use it to parallelize SymmSpMV, and compare its performance on 31 sparse matrices with other state-of-the-art coloring techniques and Intel MKL on two modern multicore processors. RACE outperforms all other approaches substantially and behaves in accordance with the Roofline model. Outliers are discussed and analyzed in detail. While we focus on SymmSpMV in this paper, our algorithm and software is applicable to any sparse matrix operation with data dependencies that can be resolved by distance-k coloring.