Abstract:Sparse matrix-vector multiplication (SpMV) plays a critical role in a wide range of linear algebra computations, particularly in scientific and engineering disciplines. However, the irregular memory access patterns, extensive memory usage, high bandwidth requirements, and underutilization of parallelism hinder the computational efficiency of SpMV on GPUs. In this paper, we propose a novel approach called block-wise dynamic mixed-precision (BDMP) to address these challenges. Our methodology involves partitioning the original matrix into uniformly sized blocks, with each block's size determined by considering architectural characteristics and accuracy requirements. Additionally, we dynamically assign precision to each block using a precision selection method that takes into account the value distribution of the original sparse matrix. We develop two distinct SpMV computation algorithms for BDMP: BDMP-PBP (Precision-based partitioning) and BDMP-TCKI (Tailored compression and kernel implementation). BDMP-PBP partitions the matrix into two independent matrices for separate computations based on block precision, offering flexibility for integration with other optimization techniques. Meanwhile, BDMP-TCKI focuses on achieving significant thread-level parallelism and memory utilization by tailoring an appropriate compressed storage format and kernel implementation for each block. We compare BDMP with NVIDIA's cuSPARSE library and three state-of-the-art SpMV methods, including SELLP, MergeBase, and BalanceCSR, using matrices from the University of Florida's SuiteSparse dataset collection. BDMP-PBP and BDMP-TCKI show average speedups up to 2.64 and 2.91 on Turing RTX 2080Ti, and up to 2.99 and 3.22 on Ampere A100. The results demonstrate that BDMP enables the optimization of computation speed without compromising the precision necessary for reliable results.

Bridging the gap between deep learning and sparse matrix format selection

MANet: an Architecture Adaptive Method for Sparse Matrix Format Selection.

Adaptive SpMV/SpMSpV on GPUs for Input Vectors of Varied Sparsity

Sparse matrix partitioning for optimizing SpMV on CPU-GPU heterogeneous platforms

Efficient Sparse Matrix Kernels based on Adaptive Workload-Balancing and Parallel-Reduction

AlphaSparse: Generating High Performance SpMV Codes Directly from Sparse Matrices

A lightweight optimization selection method for Sparse Matrix-Vector Multiplication

Optimizing Sparse Linear Algebra Through Automatic Format Selection and Machine Learning

MMSparse: 2D Partitioning of Sparse Matrix Based on Mathematical Morphology

Cascaded Prediction and Asynchronous Execution of Iterative Algorithms on Heterogeneous Platforms

Optimizing Sparse Matrix Multiplications for Graph Neural Networks

Block-wise dynamic mixed-precision for sparse matrix-vector multiplication on GPUs

A Systematic Literature Survey of Sparse Matrix-Vector Multiplication

Optimizing sparse matrix-vector multiplication based on gpu

Scale-Free Sparse Matrix-Vector Multiplication on Many-Core Architectures

Exploring the Design Space of Distributed Parallel Sparse Matrix-Multiple Vector Multiplication

Optimizing and Auto-Tuning Scale-Free Sparse Matrix-Vector Multiplication on Intel Xeon Phi

Performance Analysis and Optimization of Sparse Matrix-Vector Multiplication on Modern Multi- and Many-Core Processors

TaiChi: A Hybrid Compression Format for Binary Sparse Matrix-Vector Multiplication on GPU

Diagonally-Addressed Matrix Nicknack: How to improve SpMV performance

Sparse Matrix-Vector Multiplication Optimizations based on Matrix Bandwidth Reduction using NVIDIA CUDA