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.

$O(N)$ distributed direct factorization of structured dense matrices using runtime systems

A distributed-memory package for dense Hierarchically Semi-Separable matrix computations using randomization

An efficient multi-core implementation of a novel HSS-structured multifrontal solver using randomized sampling

Exploiting nested task-parallelism in the $\mathcal{H}-LU$ factorization

Computing Low-Rank Approximation of a Dense Matrix on Multicore CPUs with a GPU and Its Application to Solving a Hierarchically Semiseparable Linear System of Equations

Parallelization and scalability analysis of inverse factorization using the Chunks and Tasks programming model

On the Parallel I/O Optimality of Linear Algebra Kernels: Near-Optimal Matrix Factorizations

Skew-Symmetric Matrix Decompositions on Shared-Memory Architectures

Fast Structured LU Factorization for Nonsymmetric Matrices.

Accelerating Sparse Cholesky Factorization on Sunway Manycore Architecture.

Parallel Cholesky Factorization for Banded Matrices using OpenMP Tasks

GPU Accelerated Sparse Cholesky Factorization

An efficient hybrid tridiagonal divide-and-conquer algorithm on distributed memory architectures

Basker: A Threaded Sparse LU Factorization Utilizing Hierarchical Parallelism and Data Layouts

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

A High-Performance Parallel Algorithm for Nonnegative Matrix Factorization

Accelerating Mixed-Precision Out-of-Core Cholesky Factorization with Static Task Scheduling

PSL - Exploiting Parallelism, Sparsity and Locality to Accelerate Matrix Factorization on x86 Platforms.

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

On the Parallel I/O Optimality of Linear Algebra Kernels: Near-Optimal LU Factorization

A New Hybrid GPU-CPU Sparse LDLT Factorization Algorithm with GPU and CPU Factorizing Concurrently