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.

sRSP: GPUlarda Asimetrik Senkronizasyon Icin Yeni Olceklenebilir Bir Cozum

RDMA-Based Algorithms for Sparse Matrix Multiplication on GPUs

Improving the Scalability of GPU Synchronization Primitives

Simultaneous Multikernel: Fine-Grained Sharing of GPUs.

An Efficient Sequential Consistency Implementation with Dynamic Race Detection for GPUs

Simultaneous Multikernel GPU: Multi-tasking throughput processors via fine-grained sharing

A server-based approach for predictable GPU access with improved analysis

Improving GPU Performance Through Resource Sharing

Cross-Core Data Sharing for Energy-Efficient GPUs

POSTER: Accelerate GPU Concurrent Kernel Execution by Mitigating Memory Pipeline Stalls

SPGPU: Spatially Programmed GPU

RTGPU: Real-Time GPU Scheduling of Hard Deadline Parallel Tasks With Fine-Grain Utilization

Esspmv: an Embedded-FPGA-based Hardware Accelerator for Symmetric Sparse Matrix-Vector Multiplication.

Efficient Resource Sharing Through GPU Virtualization on Accelerated High Performance Computing Systems

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

RegDem: Increasing GPU Performance via Shared Memory Register Spilling

Dynamic Sampling Rate: Harnessing Frame Coherence in Graphics Applications for Energy-Efficient GPUs

ACS: Concurrent Kernel Execution on Irregular, Input-Dependent Computational Graphs

SPARTA: A Divide and Conquer Approach to Address Translation for Accelerators

LSRB-CSR: A Low Overhead Storage Format for SpMV on the GPU Systems

Efficient Hardware Accelerator Based on Medium Granularity Dataflow for SpTRSV