November 2023

Conference Paper

Progress Towards Accelerating the Unified Model on Hybrid Multi-Core Systems

By:
Zhang, Wei ; Xu, Min ; Evans, Katherine J; Norman, Matthew R; Morales Hernandez, Mario ; Mahajan, Salil
Page Number:
1-9
Issue Number:
N/A
Book Title:
Pasc '21: Proceedings of the Platform for Advanced Scientific Computing Conference
Publication Date:
November 2023
Publisher Location:
ACM, District of Columbia, United States of America
Conference Name:
Proceedings of the Platform for Advanced Scientific Computing Conference (PASC21)
Conference Location:
Geneva, Swaziland
Conference Sponsor:
Association for Computing Machinery (ACM) and the PASC structuring project
View DOI Listing:
https://doi.org/10.1145/3468267.3470612

Abstract

The cloud microphysics scheme, CASIM, and the radiation scheme, SOCRATES, are two computationally intensive parts within the Met Office's Unified Model (UM). This study enables CASIM and SOCRATES to use accelerated multi-core systems for optimal computational performance of the UM. Using profiling to guide our efforts, we refactored the code for optimal threading and kernel arrangement and implemented OpenACC directives manually or through the CLAW source-to-source translator. Initial porting results achieved 10.02x and 9.25x speedup in CASIM and SOCRATES respectively on 1 GPU compared with 1 CPU core. A granular performance analysis of the strategy and bottlenecks are discussed. These improvements will enable UM to run on heterogeneous computers and a path forward for further improvements is provided.

Related Researchers