使用 OpenMP Offload 与 Codee 优化气象预报模型
分布式、并行与集群计算
2024-09-12 v1 计算工程、金融与科学
摘要
目前，气象研究与预测模型（WRF）采用共享内存（OpenMP）和分布式内存（MPI）并行方式。为利用 NERSC Perlmutter 超级计算机上的 GPU 资源，我们将快速谱二氮微物理（FSBM）微物理方案中计算密集型例程的部分移植至 NVIDIA GPU，使用 OpenMP 设备卸载指令。为促进此过程，我们探索了一种结合运行时分析器和静态代码检查工具 Codee 来重构子程序的工作流程。我们观察到 CONUS-12km 暴雨测试用例整体加速 2.08 倍。
引用
@article{arxiv.2409.07232,
title = {Optimizing the Weather Research and Forecasting Model with OpenMP Offload and Codee},
author = {Chayanon and Wichitrnithed and Woo-Sun-Yang and Yun and He and Brad Richardson and Koichi Sakaguchi and Manuel Arenaz and William I. Gustafson and Jacob Shpund and Ulises Costi Blanco and Alvaro Goldar Dieste},
journal= {arXiv preprint arXiv:2409.07232},
year = {2024}
}