GPU-RRTMG_SW: Accelerating a Shortwave Radiative Transfer Scheme on GPU

• Published in: IEEE access Vol. 9; pp. 84231 - 84240
• Main Authors: Wang, Zhenzhen, Wang, Yuzhu, Wang, Xiaocong, Li, Fei, Zhou, Chen, Hu, Hangtian, Jiang, Jinrong
• Format: Journal Article
• Language: English
• Published: Piscataway The Institute of Electrical and Electronics Engineers, Inc. (IEEE) 2021; IEEE
• Subjects: Algorithms; Computation; compute unified device architecture; Data transmission; FORTRAN; General circulation models; graphics processing unit; Graphics processing units; High performance computing; Optimization; Radiative transfer; Short wave radiation; shortwave radiative transfer; Weather forecasting
• DOI: 10.1109/ACCESS.2021.3087507
• ISSN: 2169-3536

As high performance computing technology continues to develop rapidly, graphics processing units (GPUs) are more widely used for daily computing tasks. However, some applications, such as weather forecasting, require large-scale computation. Thus, due to the computationally intensive characteristics of the rapid radiative transfer model for general circulation models (RRTMG) in the Earth system model, this study uses GPU-related technology to accelerate computation of the model. First, two kinds of algorithms using GPU technology to accelerate the RRTMG shortwave radiation scheme (RRTMG_SW) are proposed. Then, an optimization method for data transmission between host and device is proposed. Finally, after using CUDA Fortran and CUDA C to implement these algorithms, two GPU versions of RRTMG_SW, namely CF-RRTMG_SW (CUDA Fortran version) and CC-RRTMG_SW (CUDA C version), were developed. The experimental results demonstrate that the proposed acceleration algorithms are effective. Without I/O transfer, running CC-RRTMG_SW on a NVIDIA GeForce Titan V GPU achieved a [Formula Omitted] speedup when compared to a single Intel Xeon E5-2680 CPU core.