Abstract:The International Journal of High Performance Computing Applications, Ahead of Print. We present the GPU implementation efforts and challenges of the sparse solver package STRUMPACK. The code is made publicly available on github with a permissive BSD license. STRUMPACK implements an approximate multifrontal solver, a sparse LU factorization which makes use of compression methods to accelerate time to solution and reduce memory usage. Multiple compression schemes based on rank-structured and hierarchical matrix approximations are supported, including hierarchically semi-separable, hierarchically off-diagonal butterfly, and block low rank. In this paper, we present the GPU implementation of the block low rank (BLR) compression method within a multifrontal solver. Our GPU implementation relies on highly optimized vendor libraries such as cuBLAS and cuSOLVER for NVIDIA GPUs, rocBLAS and rocSOLVER for AMD GPUs and the Intel oneAPI Math Kernel Library (oneMKL) for Intel GPUs. Additionally, we rely on external open source libraries such as SLATE (Software for Linear Algebra Targeting Exascale), MAGMA (Matrix Algebra on GPU and Multi-core Architectures), and KBLAS (KAUST BLAS). SLATE is used as a GPU-capable ScaLAPACK replacement. From MAGMA we use variable sized batched dense linear algebra operations such as GEMM, TRSM and LU with partial pivoting. KBLAS provides efficient (batched) low rank matrix compression for NVIDIA GPUs using an adaptive randomized sampling scheme. The resulting sparse solver and preconditioner runs on NVIDIA, AMD and Intel GPUs. Interfaces are available from PETSc, Trilinos and MFEM, or the solver can be used directly in user code. We report results for a range of benchmark applications, using the Perlmutter system from NERSC, Frontier from ORNL, and Aurora from ALCF. For a high frequency wave equation on a regular mesh, using 32 Perlmutter compute nodes, the factorization phase of the exact GPU solver is about 6.5× faster compared to the CPU-only solver. The BLR-enabled GPU solver is about 13.8× faster than the CPU exact solver. For a collection of SuiteSparse matrices, the STRUMPACK exact factorization on a single GPU is on average 1.9× faster than NVIDIA's cuDSS solver.

Efficient Batched CPU/GPU Implementation of Orthogonal Matching Pursuit for Python

Efficient Implementations of Orthogonal Matching Pursuit Based on Inverse Cholesky Factorization

Efficient Implementations for Orthogonal Matching Pursuit

Sparse Signals Recovery from Noisy Measurements by Orthogonal Matching Pursuit

Iterative Orthogonal Matching Pursuit and Sparse Solution

On Sparse Recovery Algorithms in Unions of Orthonormal Bases

Fast OMP for Exact Recovery and Sparse Approximation

Target Detection Sparse Algorithm by Recursive Dictionary Updating and GPU Implementation

Active Orthogonal Matching Pursuit for Sparse Subspace Clustering

Dynamic Orthogonal Matching Pursuit for Sparse Data Reconstruction

Perturbation Analysis of Orthogonal Matching Pursuit.

Random Sampling of Sparse Trigonometric Polynomials II - Orthogonal Matching Pursuit versus Basis Pursuit

Greed is Good: Algorithmic Results for Sparse Approximation

GPU Accelerated Sparse Cholesky Factorization

Simultaneous Optimized Orthogonal Matching Pursuit with Application to ECG Compression

A graphics processing unit accelerated sparse direct solver and preconditioner with block low rank compression

PERKS: a Locality-Optimized Execution Model for Iterative Memory-bound GPU Applications

Optimizing Algorithm of Sparse Linear Systems on GPU

Iterative Methods in GPU-Resident Linear Solvers for Nonlinear Constrained Optimization

Adaptive matching pursuit for sparse signal recovery

Efficient GPU Implementation for Single Block Orthogonal Dictionary Learning