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.

Parallel Triangular Solvers on GPU

Preconditioned GMRES Solver on Multiple-Gpu Architecture

Parallel optimization and application of unstructured sparse triangular solver on new generation of Sunway architecture

On Parallel Solution of Sparse Triangular Linear Systems in CUDA

A Hybrid Parallel Solving Algorithm on GPU for Quasi-Tridiagonal System of Linear Equations

Accelerating Iterative Linear Solvers Using Multiple Graphical Processing Units

Parallel ILU Preconditioners in GPU Computation

Parallel Structured Sparse Triangular Solver for GPU Platform

GPU-based Parallel Solver Via the Kantorovich Theorem for the Nonlinear Bernstein Polynomial Systems.

A Parallel Algorithm for Solving Triangular Equations under Distributed Memory Environments

GPU Accelerated Parallel Cholesky Factorization

A Parallel Iterative Method for Solving Periodical Block-Tridiagonal Linear Equations.

A Highly Efficient GPU-CPU Hybrid Parallel Implementation of Sparse LU Factorization

A Two-level GPU-Accelerated Incomplete LU Preconditioner for General Sparse Linear Systems

Parallel Solution of Triangular Systems in Direct Numerical Simulation

Parallel incomplete factorization preconditioning of block tridiagonal linear systems with 2-D domain decomposition

Accelerating Algebraic Multigrid Solvers on NVIDIA GPUs

Development of Krylov and AMG linear solvers for large-scale sparse matrices on GPUs

Parallel Tridiagonal Solver on Sunway Many-Core Processors*

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

Parallel Algorithms for Separable Elliptic Equation based on GPU