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.

Efficiency Analysis of Intel, AMD and Nvidia 64-Bit Hardware for Memory-Bound Problems: A Case Study of Ab Initio Calculations with VASP

Energy efficiency and performance analysis of a legacy atomic scale materials modeling simulator (VASP)

Numerical Performance and Throughput Benchmark for Electronic Structure Calculations in Pc-Linux Systems with New Architectures, Updated Compilers, and Libraries

Understanding HPC Benchmark Performance on Intel Broadwell and Cascade Lake Processors

Execution‐Cache‐Memory modeling and performance tuning of sparse matrix‐vector multiplication and Lattice quantum chromodynamics on A64FX

Demystifying the Performance of HPC Scientific Applications on NVM-based Memory Systems

ECM modeling and performance tuning of SpMV and Lattice QCD on A64FX

Microarchitectural comparison and in-core modeling of state-of-the-art CPUs: Grace, Sapphire Rapids, and Genoa

Towards Modeling Energy Consumption of Xeon Phi

Estimation of the workload of a hybrid computing cluster when performing modeling tasks in materials science

A High Performance Block Eigensolver for Nuclear Configuration Interaction Calculations.

Performance Analysis of CP2K Code for Ab Initio Molecular Dynamics on CPUs and GPUs

A High-Performance Implementation of Atomistic Spin Dynamics Simulations on x86 CPUs

Porting HPC Applications to AMD Instinct$^\text{TM}$ MI300A Using Unified Memory and OpenMP

Exploring and analyzing the real impact of modern on-package memory on HPC scientific kernels

Performance Analysis of CP2K Code for Ab Initio Molecular Dynamics

A Study of Performance Portability in Plasma Physics Simulations

At the Locus of Performance: Quantifying the Effects of Copious 3D-Stacked Cache on HPC Workloads

A Detailed Historical and Statistical Analysis of the Influence of Hardware Artifacts on SPEC Integer Benchmark Performance

Studying performance portability of LAMMPS across diverse GPU‐based platforms

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