BCCAGCM模式在神威·太湖之光系统的优化  

Optimizing BCCAGCM on Sunway TaihuLight

作　　者：魏敏[1] 王彬[1] 何香 孙俊[2] 姜小成[2] 肖洒[1] 张莉[3] 徐金秀[2] Wei Min;Wang Bin;He Xiang;Sun Jun;Jiang Xiaocheng;Xiao Sa;Zhang Li;Xu Jinxiu(National Meteorological Information Center,Beijing 100081;Jiangnan Institute of Computing Technology,Wuxi 214083;National Climate Center,Beijing 100081)

机构地区：[1]国家气象信息中心,北京100081 [2]江南计算技术研究所,无锡214083 [3]国家气候中心,北京100081

出　　处：《应用气象学报》2019年第4期502-512,共11页Journal of Applied Meteorological Science

基　　金：公益性行业(气象)科研专项(GYIIY201306062);国家重点研究发展计划(2016YFA0602102)

摘　　要：开展气象数值模式在神威·太湖之光系统的移植与优化,对研究模式与新型计算架构的适应性有重要意义。该文以BCCAGCM模式为研究对象,将其移植到神威·太湖之光全国产异构众核计算系统,进行性能分析,对模式动力框架和物理过程计算结构进行调整,将计算核心段采用OpenACC技术进行众核加速优化,大量代码进行算法重构。结果表明:各核心段计算效率基本达到未优化的3倍左右,最高可达14倍左右,将各核心段集成,形成异构众核集成版本,可正确、稳定运行,计算误差合理。在不同并行规模,采用从核对模式整体计算进行加速效果比较稳定,基本保持在1.9倍,26000核并行规模动力试验并行效率约70%,其他试验约为57%。With the rise of many-core processors such as Intel MIC, GPU and SW26010, the architecture of supercomputer systems has undergone great changes. The supercomputer transitions from a homogeneous system containing only multi-core CPUs to a heterogeneous system with coexistence of CPU and many-core accelerators. Heterogeneous architectures provide powerful computing power for large, complex applications. However, since the numerical model is basically based on conventional CPU development different from the many-core accelerator, the existing tens of thousands of lines of legacy code cannot take full advantage of the parallel computing capacity of the new architecture. Carrying out the porting and optimization of the weather and climate numerical model on the new system is of great significance to improve the adaptability of the model in the new computing architecture. The Sunway TaihuLight System is the world's first supercomputer with a peak performance greater than 100 PFlops based on homegrown SW26010 heterogeneous many-core chip. Each SW26010 processor consists of management processing elements (MPEs) and clusters of computing processing elements (CPEs). To support parallel computing for heterogeneous architectures, the system provides a set of compilation tools, including basic C/C++, Fortran compilers. In addition to that, there is also a customized Sunway OpenACC tool that supports the OpenACC2.0 syntax. As the atmospheric component of BCCCSM, BCCAGCM is the most computationally expensive component in typical configurations. Since BCCAGCM has not been operated in the Sunway system, BCCAGCM is first ported to the Sunway system, using only MPE to perform the computation. And then, the calculation framework is analyzed to determine the major kernels that take the most time to calculate. BCCAGCM uses a hybrid parallelization scheme combining MPI and OpenMP to complete the calculation. In the Sunway system, MPI and OpenACC are used to obtain appropriate parallelism from the CPE cluster. On one hand, by adjusting

关 键 词：BCCAGCM 神威·太湖之光 异构计算 众核 

分 类 号：P409[天文地球—大气科学及气象学]