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.

PSCToolkit: solving sparse linear systems with a large number of GPUs

On Parallel Solution of Sparse Triangular Linear Systems in CUDA

Solving unsymmetric sparse systems of linear equations with PARDISO

A Method for Solving Sparse Linear Equations of Power Systems Based on GPU

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

Alya toward exascale: algorithmic scalability using PSCToolkit

Tridiagonalization of a dense symmetric matrix on multiple GPUs and its application to symmetric eigenvalue problems

GPU Accelerated Sparse Cholesky Factorization

Evaluating Accuracy and Efficiency of HPC Solvers for Sparse Linear Systems with Applications to PDEs

Sparse Matrix-Based HPC Tomography

Optimizing Algorithm of Sparse Linear Systems on GPU

Multi-GPU aggregation-based AMG preconditioner for iterative linear solvers

Accelerating Iterative Linear Solvers Using Multiple Graphical Processing Units

Sparse tensor based nuclear gradients for periodic Hartree-Fock and low-scaling correlated wave function methods in the CP2K software package: A massively parallel and GPU accelerated implementation

Batched sparse direct solver design and evaluation in SuperLU_DIST

DBCSR: A Library for Dense Matrix Multiplications on Distributed GPU-Accelerated Systems

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

A Robust Algebraic Domain Decomposition Preconditioner for Sparse Normal Equations

PETSc/TAO Developments for GPU-Based Early Exascale Systems

GHOST: Building blocks for high performance sparse linear algebra on heterogeneous systems

A Hybrid Direct-Iterative Method for Solving KKT Linear Systems