Abstract:Sparse Triangular Solve (SpTRSV) has long been an essential kernel in the field of scientific computing. Due to its low computational intensity and internal data dependencies, SpTRSV is hard to implement and optimize on GPUs. Based on our experimental observations, existing implementations on GPUs fail to achieve the optimal performance due to their sub-optimal parallelism setups and code implementations, and lack of consideration of the irregular data distribution. Moreover, their algorithm design lacks the adaptability to different input matrices, which may involve substantial manual efforts of algorithm redesigning and parameter tuning for performance consistency. In this work, we propose AG-SpTRSV, an automatic framework to optimize SpTRSV on GPUs, which provides high performance on various matrices while eliminating the costs of manual design. AG-SpTRSV abstracts the procedures of optimizing an SpTRSV kernel as a scheme and constructs a comprehensive optimization space based on it. By defining a unified code template and preparing code variants, AG-SpTRSV enables fine-grained dynamic parallelism and adaptive code optimizations to handle various tasks. Through computation graph transformation and multi-hierarchy heuristic scheduling, AG-SpTRSV generates schemes for task partitioning and mapping, which effectively address the issues of irregular data distribution and internal data dependencies. AG-SpTRSV searches for the best scheme to optimize the target kernel for the specific matrix. A learned lightweight performance model is also introduced to reduce search costs and provide an efficient end-to-end solution. Experimental results with SuiteSparse Matrix Collection on NVIDIA Tesla A100 and RTX 3080 Ti show that AG-SpTRSV outperforms state-of-the-art implementations with geometric average speedups of 2.12x ∼ 3.99x. With the performance model enabled, AG-SpTRSV can provide an efficient end-to-end solution, with preprocessing times ranging from 3.4 to 245 times of the execution time.

Sgap: Towards Efficient Sparse Tensor Algebra Compilation for GPU

A High-Performance Sparse Tensor Algebra Compiler in Multi-Level IR

A Heterogeneous Parallel Computing Approach Optimizing SpTTM on CPU-GPU Via GCN

Atomic Reduction Based Sparse Matrix-Transpose Vector Multiplication on GPUs

AG-SpTRSV: an Automatic Framework to Optimize Sparse Triangular Solve on GPUs

Compilation of Modular and General Sparse Workspaces

SPLAT: A framework for optimised GPU code-generation for SParse reguLar ATtention

FEASTA: A Flexible and Efficient Accelerator for Sparse Tensor Algebra in Machine Learning

Optimizing sparse matrix-vector multiplication based on gpu

A Novel Parallel Algorithm for Sparse Tensor Matrix Chain Multiplication via TCU-Acceleration

Efficient Utilization of Multi-Threading Parallelism on Heterogeneous Systems for Sparse Tensor Contraction

EC-SpMM: Efficient Compilation of SpMM Kernel on GPUs.

TSTC: Enabling Efficient Training Via Structured Sparse Tensor Compilation

SpMMPlu: A Compiler Plug-in with Sparse IR for Efficient Sparse Matrix Multiplication.

Performance Optimization for Sparse A(T)Ax in Parallel on Multicore Cpu

A Survey of Accelerating Parallel Sparse Linear Algebra

Optimizing Sparse Tensor Times Matrix on GPUs

A new segmentation-based GPU-accelerated sparse matrix-vector multiplication

TileSpMSpV: A Tiled Algorithm for Sparse Matrix-Sparse Vector Multiplication on GPUs

Distributed-Memory Parallel Algorithms for Sparse Matrix and Sparse Tall-and-Skinny Matrix Multiplication

FlashSparse: Minimizing Computation Redundancy for Fast Sparse Matrix Multiplications on Tensor Cores