Kilometer-Level Coupled Modeling Using 40 Million Cores: An Eight-Year Journey of Model Development

• Main Authors: Duan, Xiaohui, Li, Yuxuan, Liu, Zhao, Yang, Bin, Zheng, Juepeng, Fu, Haohuan, Zhang, Shaoqing, Xu, Shiming, Gao, Yang, Xue, Wei, Wei, Di, Lv, Xiaojing, Yan, Lifeng, Huang, Haopeng, Lu, Haitian, Wan, Lingfeng, Lin, Haoran, Chang, Qixin, Li, Chenlin, He, Quanjie, Song, Zeyu, Wang, Xuantong, Yu, Yangyang, Fan, Xilong, Qu, Zhaopeng, Xu, Yankun, Guo, Xiuwen, Fei, Yunlong, Wang, Zhaoying, Li, Mingkui, Jiang, Yingjing, Lu, Lv, Su, Liang, Fu, Jiayu, Yu, Peinan, Liu, Weiguo, Wu, Lixin, Wang, Lanning, Liu, Xin, Chen, Dexun, Yang, Guangwen
• Format: Journal Article
• Language: English
• Published: 15.04.2024
• Subjects: Computer Science - Distributed, Parallel, and Cluster Computing
• DOI: 10.48550/arxiv.2404.10253

With current and future leading systems adopting heterogeneous architectures, adapting existing models for heterogeneous supercomputers is of urgent need for improving model resolution and reducing modeling uncertainty. This paper presents our three-week effort on porting a complex earth system model, CESM 2.2, to a 40-million-core Sunway supercomputer. Taking a non-intrusive approach that tries to minimizes manual code modifications, our project tries to achieve both improvement of performance and consistency of the model code. By using a hierarchical grid system and an OpenMP-based offloading toolkit, our porting and parallelization effort covers over 80% of the code, and achieves a simulation speed of 340 SDPD (simulated days per day) for 5-km atmosphere, 265 SDPD for 3-km ocean, and 222 SDPD for a coupled model, thus making multi-year or even multi-decadal experiments at such high resolution possible.