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.

GPU Implementation of a Sophisticated Implicit Low-Order Finite Element Solver with FP21-32-64 Computation Using OpenACC

Parallelized Implementation of the Finite Particle Method for Explicit Dynamics in GPU

Implicit Low-Order Unstructured Finite-Element Multiple Simulation Enhanced by Dense Computation using OpenACC

OpenACC offloading of the MFC compressible multiphase flow solver on AMD and NVIDIA GPUs

Massive parallelization and performance enhancement of an immersed boundary method based unsteady flow solver

Generalized Gpu Acceleration For Applications Employing Finite-Volume Methods

Accelerating an Iterative Eigensolver for Nuclear Structure Configuration Interaction Calculations on GPUs Using OpenACC

OpenACC acceleration of an unstructured CFD solver based on a reconstructed discontinuous Galerkin method for compressible flows

Efficient and portable acceleration of quantum chemical many-body methods in mixed floating point precision using OpenACC compiler directives

Method for portable, scalable, and performant GPU-accelerated simulation of multiphase compressible flow

An adaptive finite element multigrid solver using GPU acceleration

Unified schemes for directive-based GPU offloading

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

Performance Portable Graphics Processing Unit Acceleration of a High-Order Finite Element Multiphysics Application

Method for scalable and performant GPU-accelerated simulation of multiphase compressible flow

Optimization and Portability of a Fusion OpenACC-based FORTRAN HPC Code from NVIDIA to AMD GPUs

Multi-GPU Performance Optimization of a CFD Code using OpenACC on Different Platforms

Evaluating performance portability of five shared-memory programming models using a high-order unstructured CFD solver

Accelerated FDPS — Algorithms to Use Accelerators with FDPS

Optimizing Finite Volume Method Solvers on Nvidia GPUs.

Portability and Scalability of OpenMP Offloading on State-of-the-art Accelerators