Back

Minisymposium Presentation

Porting and Optimizing Momentum, CASIM and SOCRATES for GPU Architectures

Tuesday, June 4, 2024
16:30
-
17:00
CEST
Climate, Weather and Earth Sciences
Climate, Weather and Earth Sciences
Climate, Weather and Earth Sciences
Chemistry and Materials
Chemistry and Materials
Chemistry and Materials
Computer Science and Applied Mathematics
Computer Science and Applied Mathematics
Computer Science and Applied Mathematics
Humanities and Social Sciences
Humanities and Social Sciences
Humanities and Social Sciences
Engineering
Engineering
Engineering
Life Sciences
Life Sciences
Life Sciences
Physics
Physics
Physics

Presenter

Wei
Zhang
-
Oak Ridge National Laboratory

Wei Zhang is a research scientist in the Computational Earth Sciences Group of the Computational Sciences and Engineering Division at Oak Ridge National Laboratory. She received her Ph.D in Atmospheric Science at Stony Brook University in 2018, and her MS and BS degrees in Atmospheric Science from Peking University. Her major research includes cloud microphysics, cloud-aerosol interaction, their representation in atmospheric models, HPC techniques and GPU computing technology. After joining ORNL, she has contributed to several projects on model optimization and fog science research.

Description

Exploiting GPUs is both an opportunity and a challenge for weather and climate codes. They present an opportunity as the massive parallelism they possess can allow these codes to achieve very high computational performance. They present a challenge as exploiting this parallelism can require the refactoring of many thousands of lines of science code and a new programming model from existing CPU code bases. In this presentation, we will describe the development of the GPU-enabled cloud microphysics scheme - CASIM and radiation scheme - SOCRATES, and a Domain Specific Language the Met Office is using to achieve performance portability for its new weather and climate model, LFRic and the wider modelling system known as Momentum. We will show how the Met Office is using PSyclone, a Domain Specific Compiler to keep single source science code whilst targeting multiple programming models for different processor architectures. The presentation will conclude the strategy and progress for porting and optimizing Momentum for GPUs, and how PSyclone follows the porting experience of ORNL to optimize CASIM and SOCRATES on GPUs.

Authors