Porting WarpX to GPU-accelerated platforms

A. Myers , A. Almgren , L.D. Amorim , J. Bell , L. Fedeli (1, 2) , L. Ge , K. Gott , D.P. Grote , M. Hogan , A. Huebl , R. Jambunathan , R. Lehe , C. Ng , M. Rowan , O. Shapoval , M. Thévenet , J.-L. Vay , H. Vincenti (1, 2) , E. Yang , N. Zaïm (1, 2) , W. Zhang , Y. Zhao , E. Zoni

1 LIDYL - Laboratoire Interactions, Dynamiques et Lasers (ex SPAM)
2 PHI - Physique à Haute Intensité

WarpX is a general purpose electromagnetic particle-in-cell code that was originally designed to run on many-core CPU architectures. We describe the strategy followed to allow WarpX to use the GPU-accelerated nodes on OLCF's Summit supercomputer, a strategy we believe will extend to the upcoming machines Frontier and Aurora. We summarize the challenges encountered, lessons learned, and give current performance results on a series of relevant benchmark problems.

Mots clés: Exascale Computing; Particle-in-cell methods; Accelerator modeling

Domaines: Informatique [cs]; Physique des accélérateurs [physics.acc-ph]

A. Myers, A. Almgren, L.D. Amorim, J. Bell, L. Fedeli, et al.. Porting WarpX to GPU-accelerated platforms. Parallel Comput., 2021, 108, pp.102833. ⟨10.1016/j.parco.2021.102833⟩. ⟨hal-03136712⟩