Accelerating Numerical Relativity with Code Generation: CUDA-enabled Hyperbolic Relaxation

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Tootle, Samuel D., Werneck, Leonardo R., Assumpção, Thiago, Jacques, Terrence Pierre, Etienne, Zachariah B.
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866915292256403456
author Tootle, Samuel D.
Werneck, Leonardo R.
Assumpção, Thiago
Jacques, Terrence Pierre
Etienne, Zachariah B.
author_facet Tootle, Samuel D.
Werneck, Leonardo R.
Assumpção, Thiago
Jacques, Terrence Pierre
Etienne, Zachariah B.
contents Next-generation gravitational wave detectors such as Cosmic Explorer, the Einstein Telescope, and LISA, demand highly accurate and extensive gravitational wave (GW) catalogs to faithfully extract physical parameters from observed signals. However, numerical relativity (NR) faces significant challenges in generating these catalogs at the required scale and accuracy on modern computers, as NR codes do not fully exploit modern GPU capabilities. In response, we extend NRPy, a Python-based NR code-generation framework, to develop NRPyEllipticGPU -- a CUDA-optimized elliptic solver tailored for the binary black hole (BBH) initial data problem. NRPyEllipticGPU is the first GPU-enabled elliptic solver in the NR community, supporting a variety of coordinate systems and demonstrating substantial performance improvements on both consumer-grade and HPC-grade GPUs. We show that, when compared to a high-end CPU, NRPyEllipticGPU achieves on a high-end GPU up to a sixteenfold speedup in single precision while increasing double-precision performance by a factor of 2--4. This performance boost leverages the GPU's superior parallelism and memory bandwidth to achieve a compute-bound application and enhancing the overall simulation efficiency. As NRPyEllipticGPU shares the core infrastructure common to NR codes, this work serves as a practical guide for developing full, CUDA-optimized NR codes.
format Preprint
id arxiv_https___arxiv_org_abs_2501_14030
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Accelerating Numerical Relativity with Code Generation: CUDA-enabled Hyperbolic Relaxation
Tootle, Samuel D.
Werneck, Leonardo R.
Assumpção, Thiago
Jacques, Terrence Pierre
Etienne, Zachariah B.
General Relativity and Quantum Cosmology
Computational Physics
Next-generation gravitational wave detectors such as Cosmic Explorer, the Einstein Telescope, and LISA, demand highly accurate and extensive gravitational wave (GW) catalogs to faithfully extract physical parameters from observed signals. However, numerical relativity (NR) faces significant challenges in generating these catalogs at the required scale and accuracy on modern computers, as NR codes do not fully exploit modern GPU capabilities. In response, we extend NRPy, a Python-based NR code-generation framework, to develop NRPyEllipticGPU -- a CUDA-optimized elliptic solver tailored for the binary black hole (BBH) initial data problem. NRPyEllipticGPU is the first GPU-enabled elliptic solver in the NR community, supporting a variety of coordinate systems and demonstrating substantial performance improvements on both consumer-grade and HPC-grade GPUs. We show that, when compared to a high-end CPU, NRPyEllipticGPU achieves on a high-end GPU up to a sixteenfold speedup in single precision while increasing double-precision performance by a factor of 2--4. This performance boost leverages the GPU's superior parallelism and memory bandwidth to achieve a compute-bound application and enhancing the overall simulation efficiency. As NRPyEllipticGPU shares the core infrastructure common to NR codes, this work serves as a practical guide for developing full, CUDA-optimized NR codes.
title Accelerating Numerical Relativity with Code Generation: CUDA-enabled Hyperbolic Relaxation
topic General Relativity and Quantum Cosmology
Computational Physics
url https://arxiv.org/abs/2501.14030