Toward the modeling of chains of plasma accelerator stages with WarpX

Abstract

The Particle-In-Cell code WarpX is being developed by a team of the U.S. DOE Exascale Computing Project to enable the modeling of chains of tens of plasma accelerators on exascale supercomputers, for future collider designs. The code is combining the latest algorithmic advances (e.g., boosted frame, pseudo-spectral Maxwell solvers) with mesh refinement and runs on the latest CPU and GPU architectures. An example of the application to the modeling of up to three successive muti-GeV stages is presented. The latest implementation on GPU architectures is also reported.

A hybrid nodal-staggered pseudo-spectral electromagnetic particle-in-cell method with finite-order centering

“…This section presents various physics applications to test the novel hybrid scheme, with the different PSATD PIC algorithms described in Section 4. All simulations and results have been performed and obtained with the open-source electromagnetic PIC code WarpX [47,48,49].…”

Modeling of a chain of three plasma accelerator stages with the WarpX electromagnetic PIC code on GPUs

“…[10][11][12] All computer operations in WarpX are implemented in the aforementioned manner, which enables complete GPU acceleration and avoids host-device data transfers for the critical path of the application. 13 Unnoticed by the WarpX developer, AMReX's low-level, onnode primitives are implemented with accelerated programing models, most prominently OpenMP 3.1þ for CPU and CUDA 9.0þ for Nvidia GPU support. Consequently, ongoing activities that will enable Intel GPU support (through SYCL/DPCþþ 12 ) and AMD GPU support (through HIP 14 ) are causing little to no change to numerical algorithms implemented in WarpX.…”

Particle-in-cell Simulations of Relativistic Magnetic Reconnection with Advanced Maxwell Solver Algorithms

“…The PIC algorithm captures reconnection physics accurately from first principles, but it can be computationally expensive, especially when performing high-resolution simulations with higher-order particle shape factors. Graphics processing units (GPUs) can offer remarkable acceleration over conventional CPU architectures for a number of scientific applications, including PIC (Bussmann et al 2013;Germaschewski et al 2016;Chien et al 2020;Vay et al 2020;Myers et al 2021). We use the GPU-accelerated electromagnetic PIC code, WarpX (Vay et al 2020;Myers et al 2021).…”

“…Graphics processing units (GPUs) can offer remarkable acceleration over conventional CPU architectures for a number of scientific applications, including PIC (Bussmann et al 2013;Germaschewski et al 2016;Chien et al 2020;Vay et al 2020;Myers et al 2021). We use the GPU-accelerated electromagnetic PIC code, WarpX (Vay et al 2020;Myers et al 2021). It has excellent fullmachine scaling at leadership-class computing facilities, including Summit and Perlmutter (NVIDIA GPUs) and the world's first reported exascale machine, Frontier (AMD GPUs;Fedeli et al 2022).…”