spECK: accelerating GPU sparse matrix-matrix multiplication through lightweight analysis
Mathias Parger, Martin Winter, Daniel Mlakar, Markus Steinberger
Abstract
Sparse general matrix-matrix multiplication on GPUs is challenging due to the varying sparsity patterns of sparse matrices. Existing solutions achieve good performance for certain types of matrices, but fail to accelerate all kinds of matrices in the same manner. Our approach combines multiple strategies with dynamic parameter selection to dynamically choose and tune the best fitting algorithm for each row of the matrix. This choice is supported by a lightweight, multi-level matrix analysis, which carefully balances analysis cost and expected performance gains. Our evaluation on thousands of matrices with various characteristics shows that we outperform all currently available solutions in 79% over all matrices with >15k products and that we achieve the second best performance in 15%. For these matrices, our solution is on average 83% faster than the second best approach and up to 25X faster than other state-of-the-art GPU implementations. Using our approach, applications can expect great performance independent of the matrices they work on.
Ask about this paper
Ask your agent about it.
Lune has read the top-tier papers around this one, so every answer names the papers it rests on.
Cited by top-tier papers3
- Understanding and bridging the gaps in current GNN performance optimizationsKezhao Huang, Jidong Zhai, Zhen Zheng, Youngmin Yi et al.PPoPP 2021 · 87 citations
- TileSpGEMM: a tiled algorithm for parallel sparse general matrix-matrix multiplication on GPUsYuyao Niu, Zhengyang Lu, Haonan Ji, Shuhui Song et al.PPoPP 2022 · 66 citations
- Compilation of Modular and General Sparse WorkspacesGenghan Zhang, Olivia Hsu, Fredrik KjolstadPLDI 2024 · 6 citations
Related papers
- SPAGHETTI: Streaming Accelerators for Highly Sparse GEMM on FPGAsReza Hojabr, Ali Sedaghati, Amirali Sharifian, Ahmad Khonsari et al.HPCA 2021 · 66 citations
- Efficient Algorithm Design of Optimizing SpMV on GPUGenshen Chu, Yuanjie He, Lingyu Dong, Zhezhao Ding et al.HPDC 2023 · 17 citations
- GPUs All Grown-Up: Fully Device-Driven SpMV Using GPU Work GraphsFabian Wildgrube, Pete Ehrett, Paul Trojahn, Richard Membarth et al.ISCA 2025 · 3 citations
- Heuristic adaptability to input dynamics for SpMM on CPUsGuohao Dai, Guyue Huang, Shang Yang, Zhongming Yu et al.DAC 2022 · 27 citations
- Spada: Accelerating Sparse Matrix Multiplication with Adaptive DataflowZhiyao Li, Jiaxiang Li, Taijie Chen, Dimin Niu et al.ASPLOS 2023 · 59 citations
