Presentation
Kilometer-Scale AI-Powered and Performance-Portable Earth System Model (AP3ESM) to Achieve Year-Scale Simulation Speed on Heterogeneous Supercomputers
DescriptionKilometer-scale Earth system models (ESMs) necessitate exascale supercomputers to facilitate realistic simulations of weather phenomena and climate variability over a time span ranging from days to decades. We present AP3ESM, an ultra-high-resolution, AI-Powered, Performance-Portable ESM coupling atmosphere, land surface, ocean, and sea ice components. By leveraging the performance portability features of Kokkos and OpenMP, the AP3ESM operates efficiently on two heterogeneous systems while incurring minimal development overhead. Advanced optimization techniques, such as adaptive parallel algorithms, AI-enhanced physical parameterizations, and mixed-precision computations, have been implemented to further boost the computational efficiency. The standalone atmospheric and oceanic components of AP3ESM each reach 1-km resolution, attaining 0.60 and 1.98 simulated-years-per-day (SYPD) using 34.1 million cores and 16,085 GPUs, respectively. The holistic AP3ESM (3-km atmosphere, 2-km ocean) sustains 1.01 SYPD on 36.6 million cores. Notably, the forecast experiment successfully captures Super Typhoon Doksuri in 2023 and its associated extreme rainfall across China.
Authors
Event Type
Paper
TimeThursday, 20 November 20252:36pm - 2:58pm CST
Location263-264
Applications
GBC