Abstract:The exascale race is at an end with the announcement of the Aurora and Frontier machines. This next generation of supercomputers utilize diverse hardware architectures to achieve their compute performance, providing an added onus on the performance portability of applications. An expanding fragmentation of programming models would provide a compounding optimization challenge were it not for the evolution of performance-portable frameworks, providing unified models for mapping abstract hierarchies of parallelism to diverse architectures. A solution to this challenge is the evolution of performance-portable frameworks, providing unified models for mapping abstract hierarchies of parallelism to diverse architectures. Kokkos is one such performance portable programming model for C++ applications, providing back-end implementations for each major HPC platform. Even with a performance portable framework, restructuring algorithms to expose higher degrees of parallelism is non-trivial. The Spectral Neighbor Analysis Potential (SNAP) is a machine-learned inter-atomic potential utilized in cutting-edge molecular dynamics simulations. Previous implementations of the SNAP calculation showed a downward trend in their performance relative to peak on newer-generation CPUs and low performance on GPUs. In this paper we describe the restructuring and optimization of SNAP as implemented in the Kokkos CUDA backend of the LAMMPS molecular dynamics package, benchmarked on NVIDIA GPUs. We identify novel patterns of hierarchical parallelism, facilitating a minimization of memory access overheads and pushing the implementation into a compute-saturated regime. Our implementation via Kokkos enables recompile-and-run efficiency on upcoming architectures. We find a $\sim$22x time-to-solution improvement relative to an existing implementation as measured on an NVIDIA Tesla V100-16GB for an important benchmark.

Optimization and Evaluation of VLPL-S on Knights Landing

Optimizing A Particle-In-Cell Code On Intel Knights Landing

Co-design of a Particle-in-Cell Plasma Simulation Code for Intel Xeon Phi: A First Look at Knights Landing

Parallelization and Optimization of Laser-Plasma-Interaction Simulation

Analysis and MPI Implementation of LQCD Dslash on Sunway TaihuLight*

Parallelization and I/O Optimization for a 3-D Plasma Simulation Code

A Particle-In-Cell Code Comparison for Ion Acceleration: EPOCH, LSP, and WarpX

High Performance Optimizations For Nuclear Physics Code Mfdn On Knl

Exploiting Parallelism on Shared Memory in the QED Particle-in-Cell Code PICADOR with Greedy Load Balancing

The Development of Laser-Plasma Interaction Program Lap3d on Thousands of Processors

QuickPIC: a highly efficient fully parallelized PIC code for plasma-based acceleration

A NVIDA kepler based acceleration of PIC method

VPIC 2.0: Next Generation Particle-in-Cell Simulations

Performance Analysis and Application of 2-D Plasma Parallel Particle-in-cell Simulations

Porting the grid-based 3D+3V hybrid-Vlasov kinetic plasma simulation Vlasiator to heterogeneous GPU architectures

The Plasma Simulation Code: A modern particle-in-cell code with load-balancing and GPU support

Localized On-Chip Power Delivery Network Optimization via Sequence of Linear Programming

Rapid Exploration of Optimization Strategies on Advanced Architectures using TestSNAP and LAMMPS

Optimizing High-Performance Linpack for Exascale Accelerated Architectures

MILC Code Performance on High End CPU and GPU Supercomputer Clusters

PM2D: A Parallel GPU-based Code for the Kinetic Simulation of Laser Plasma Instabilities at Large Scales