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.

Optimized Speculative Sampling for GPU Hardware Accelerators

Turbo: Opportunistic Enhancement for Edge Video Analytics

Leveraging Speculative Sampling and KV-Cache Optimizations Together for Generative AI using OpenVINO

BASS: Batched Attention-optimized Speculative Sampling

Accelerating Large Language Model Decoding with Speculative Sampling

Accelerating Stratified Sampling SGD by Reconstructing Strata

SpecTr: Fast Speculative Decoding via Optimal Transport

A Preliminary Study on Accelerating Simulation Optimization with GPU Implementation

A Speculative HMMER Search Implementation on GPU

ACC Saturator: Automatic Kernel Optimization for Directive-Based GPU Code

Parallelism Vs. Speculation

Optimizing GPU-based Graph Sampling and Random Walk for Efficiency and Scalability

Accelerating Diffusion Sampling with Optimized Time Steps

Heuristic Adaptability to Input Dynamics for SpMM on GPUs

Revisiting Approximate Query Processing and Bootstrap Error Estimation on GPU

Faster Sampling via Stochastic Gradient Proximal Sampler

Fast Stochastic Composite Minimization and an Accelerated Frank-Wolfe Algorithm under Parallelization

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

Automatic source code generation for deterministic global optimization with parallel architectures

Diffusion Sampling Correction via Approximately 10 Parameters

SIP: Autotuning GPU Native Schedules via Stochastic Instruction Perturbation