Leveraging MLIR for GPU-accelerated stencil computing, by Jean-Michel Gorius

Leveraging MLIR for GPU-accelerated stencil computing, by Jean-Michel Gorius Présentation réalisée dans le cadre de la journée MLIR: The Good, the Bad and the Ugly Plus d'information sur https://www.gdr-soc.cnrs.fr/2023/07/1... Weather and climate models make extensive use of stencil computations to model physical phenomena. These models have grown in complexity over the past decades and make up large parts of today’s HPC workloads. With new architectures switching to heterogeneous compute nodes, climate models need to be compiled and optimized for accelerator hardware. In this presentation, we illustrate how we use the MLIR infrastructure to compile stencil code for GPU accelerators. We illustrate how we leverage the multi-level capabilities of MLIR by representing stencils in a high-level intermediate representation and subsequently lower this representation down to GPU code. We show that our toolchain outperforms the state-of-the-art when compiling the dynamical core of the European weather and climate model.