Assessing the robustness and scalability of the accelerated pseudo-transient method

Ludovic Räss (1, 2) , Ivan Utkin (3, 2) , Thibault Duretz (4, 5) , Samuel Omlin (6) , Yuri Y Podladchikov (3, 7, 8)

1 Laboratory of Hydraulics, Hydrology and Glaciology (VAW)
2 Swiss Federal Institute for Forest, Snow and Landscape Research WSL
3 MSU - Lomonosov Moscow State University
4 GR - Géosciences Rennes
5 Goethe-Universität Frankfurt am Main
6 CSCS - Swiss National Supercomputing Centre-SCR department
7 UNIL - Université de Lausanne = University of Lausanne
8 Swiss Geocomputing Centre

The development of highly efficient, robust and scalable numerical algorithms lags behind the rapid increase in massive parallelism of modern hardware. We address this challenge with the accelerated pseudo-transient (PT) iterative method and present a physically motivated derivation. We analytically determine optimal iteration parameters for a variety of basic physical processes and confirm the validity of theoretical predictions with numerical experiments. We provide an efficient numerical implementation of PT solvers on graphical processing units (GPUs) using the Julia language. We achieve a parallel efficiency of more than 96 % on 2197 GPUs in distributed-memory parallelisation weak-scaling benchmarks. The 2197 GPUs allow for unprecedented tera-scale solutions of 3D variable viscosity Stokes flow on 49953 grid cells involving over 1.2 trillion degrees of freedom (DoFs). We verify the robustness of the method by handling contrasts up to 9 orders of magnitude in material parameters such as viscosity and arbitrary distribution of viscous inclusions for different flow configurations. Moreover, we show that this method is well suited to tackle strongly nonlinear problems such as shear-banding in a visco-elasto-plastic medium. A GPU-based implementation can outperform direct-iterative solvers based on central processing units (CPUs) in terms of wall time, even at relatively low spatial resolution. We additionally motivate the accessibility of the method by its conciseness, flexibility, physically motivated derivation and ease of implementation. This solution strategy thus has a great potential for future high-performance computing (HPC) applications, and for paving the road to exascale in the geosciences and beyond.

Domaines: Informatique [cs]

Ludovic Räss, Ivan Utkin, Thibault Duretz, Samuel Omlin, Yuri Y Podladchikov. Assessing the robustness and scalability of the accelerated pseudo-transient method. Geoscientific Model Development, 2022, 15 (14), pp.5757 - 5786. ⟨10.5194/gmd-15-5757-2022⟩. ⟨insu-03752043⟩