Performance-Portable Numerical Relativity with AthenaK

Abstract

We present the numerical relativity module within AthenaK, an open source performance-portable astrophysics code designed for exascale computing applications. This module employs the Z4c formulation to solve the Einstein equations. We demonstrate its accuracy through a series of standard numerical relativity tests, including convergence of the gravitational waveform from binary black hole coalescence. Furthermore, we conduct scaling tests on OLCF Frontier and NERSC Perlmutter, where AthenaK exhibits excellent weak scaling efficiency of 80% on up to 65,536 AMD MI250X GPUs on Frontier (relative to 4 GPUs) and strong scaling efficiencies of 84% and 77% on AMD MI250X and NVIDIA A100 GPUs on Frontier and Perlmutter respectively. Additionally, we observe a significant performance boost, with two orders of magnitude speedup ($\gtrsim 200\times$) on a GPU compared to a single CPU core, affirming that AthenaK is well-suited for exascale computing, thereby expanding the potential for breakthroughs in numerical relativity research.