Saved in:
Bibliographic Details
Main Authors: Vembe, Johanne Elise, Krotkiewski, Marcin, Bjørgve, Magnar, Førre, Morten, Agueny, Hicham
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2512.21697
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909975964549120
author Vembe, Johanne Elise
Krotkiewski, Marcin
Bjørgve, Magnar
Førre, Morten
Agueny, Hicham
author_facet Vembe, Johanne Elise
Krotkiewski, Marcin
Bjørgve, Magnar
Førre, Morten
Agueny, Hicham
contents Modern heterogeneous high-performance computing (HPC) systems powered by advanced graphics processing unit (GPU) architectures enable accelerating computing with unprecedented performance and scalability. Here, we present a GPU-accelerated solver for the three-dimensional (3D) time-dependent Dirac equation optimized for distributed HPC systems. The solver named GaDE is designed to simulate the electron dynamics in atoms induced by electromagnetic fields in the relativistic regime. It combines MPI with CUDA/HIP to target both NVIDIA and AMD GPU architectures. We discuss our implementation strategies in which most of the computations are carried out on GPUs, taking advantage of the GPU-aware MPI feature to optimize communication performance. We evaluate GaDE on the pre-exascale supercomputer LUMI, powered by AMD MI250X GPUs and HPE's Slingshot interconnect. Single-GPU performance on NVIDIA A100, GH200, and AMD MI250X shows comparable performance between A100 and MI250X in compute and memory bandwidth, with GH200 delivering higher performance. Weak scaling on LUMI demonstrates exceptional scalability, achieving 85% parallel efficiency across 2048 GPUs, while strong scaling delivers a 16x speedup on 32 GPUs - 50% efficiency for a communication-intensive, time-dependent Dirac equation solver. These results demonstrate GaDE's high scalability, making it suitable for exascale systems and enabling predictive simulations for ultra-intense laser experiments probing relativistic quantum effects.
format Preprint
id arxiv_https___arxiv_org_abs_2512_21697
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle GaDE -- GPU-acceleration of time-dependent Dirac Equation for exascale
Vembe, Johanne Elise
Krotkiewski, Marcin
Bjørgve, Magnar
Førre, Morten
Agueny, Hicham
Computational Physics
Quantum Physics
G.4; G.1; D.3; I.6; J.2
Modern heterogeneous high-performance computing (HPC) systems powered by advanced graphics processing unit (GPU) architectures enable accelerating computing with unprecedented performance and scalability. Here, we present a GPU-accelerated solver for the three-dimensional (3D) time-dependent Dirac equation optimized for distributed HPC systems. The solver named GaDE is designed to simulate the electron dynamics in atoms induced by electromagnetic fields in the relativistic regime. It combines MPI with CUDA/HIP to target both NVIDIA and AMD GPU architectures. We discuss our implementation strategies in which most of the computations are carried out on GPUs, taking advantage of the GPU-aware MPI feature to optimize communication performance. We evaluate GaDE on the pre-exascale supercomputer LUMI, powered by AMD MI250X GPUs and HPE's Slingshot interconnect. Single-GPU performance on NVIDIA A100, GH200, and AMD MI250X shows comparable performance between A100 and MI250X in compute and memory bandwidth, with GH200 delivering higher performance. Weak scaling on LUMI demonstrates exceptional scalability, achieving 85% parallel efficiency across 2048 GPUs, while strong scaling delivers a 16x speedup on 32 GPUs - 50% efficiency for a communication-intensive, time-dependent Dirac equation solver. These results demonstrate GaDE's high scalability, making it suitable for exascale systems and enabling predictive simulations for ultra-intense laser experiments probing relativistic quantum effects.
title GaDE -- GPU-acceleration of time-dependent Dirac Equation for exascale
topic Computational Physics
Quantum Physics
G.4; G.1; D.3; I.6; J.2
url https://arxiv.org/abs/2512.21697