Semi-Lagrangian 4d, 5d, and 6d kinetic plasma simulation on large scale GPU equipped supercomputer

Abstract

Running kinetic plasma physics simulations using grid-based solvers is very demanding both in terms of memory as well as computational cost. This is primarily due to the up to six-dimensional phase space and the associated unfavorable scaling of the computational cost as a function of grid spacing (often termed the curse of dimensionality). In this paper, we present 4d, 5d, and 6d simulations of the Vlasov--Poisson equation with a split-step semi-Lagrangian discontinuous Galerkin scheme on graphic processing units (GPUs). The local communication pattern of this method allows an efficient implementation on large-scale GPU-based systems and emphasizes the importance of considering algorithmic and high-performance computing aspects in unison. We demonstrate a single node performance above 2 TB/s effective memory bandwidth (on a node with 4 A100 GPUs) and show excellent scaling (parallel efficiency between 30% and 67%) for up to 1536 A100 GPUs on JUWELS Booster.

arXiv commentSubmitted to The International Journal of High Performance Computing Applications

References in corpus (7)

• Kinetic Simulations of Magnetized Turbulence in Astrophysical Plasmas
• A Hamiltonian splitting for the Vlasov-Maxwell system
• A mass, momentum, and energy conservative dynamical low-rank scheme for the Vlasov equation
• Convergence analysis of Strang splitting for Vlasov-type equations
• A comparison of semi-Lagrangian discontinuous Galerkin and spline based Vlasov solvers in four dimensions
• A portable platform for accelerated PIC codes and its application to GPUs using OpenACC
• A strategy to suppress recurrence in grid-based Vlasov solvers