Performance-Portable Numerical Relativity with AthenaK

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Zhu, Hengrui, Fields, Jacob, Zappa, Francesco, Radice, David, Stone, James, Rashti, Alireza, Cook, William, Bernuzzi, Sebastiano, Daszuta, Boris
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914950102908928
author Zhu, Hengrui
Fields, Jacob
Zappa, Francesco
Radice, David
Stone, James
Rashti, Alireza
Cook, William
Bernuzzi, Sebastiano
Daszuta, Boris
author_facet Zhu, Hengrui
Fields, Jacob
Zappa, Francesco
Radice, David
Stone, James
Rashti, Alireza
Cook, William
Bernuzzi, Sebastiano
Daszuta, Boris
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2409_10383
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Performance-Portable Numerical Relativity with AthenaK
Zhu, Hengrui
Fields, Jacob
Zappa, Francesco
Radice, David
Stone, James
Rashti, Alireza
Cook, William
Bernuzzi, Sebastiano
Daszuta, Boris
General Relativity and Quantum Cosmology
High Energy Astrophysical Phenomena
Instrumentation and Methods for Astrophysics
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.
title Performance-Portable Numerical Relativity with AthenaK
topic General Relativity and Quantum Cosmology
High Energy Astrophysical Phenomena
Instrumentation and Methods for Astrophysics
url https://arxiv.org/abs/2409.10383