Compute unified device architecture (CUDA)-based parallelization of WRF Kessler cloud microphysics scheme

摘要

In recent years, graphics processing units (CPUs) have emerged as a low-cost, low-power and a very high performance alternative to conventional central processing units (CPUs). The latest CPUs offer a speedup of two-to-three orders of magnitude over CPU for various science and engineering applications. The Weather Research and Forecasting (WRF) model is the latest-generation numerical weather prediction model. It has been designed to serve both operational forecasting and atmospheric research needs. It proves useful for a broad spectrum of applications for domain scales ranging from meters to hundreds of kilometers. WRF computes an approximate solution to the differential equations which govern the air motion of the whole atmosphere. Kessler microphysics module in WRF is a simple warm cloud scheme that includes water vapor, cloud water and rain. Microphysics processes which are modeled are rain production, fall and evaporation. The accretion and auto-conversion of cloud water processes are also included along with the production of cloud water from condensation. In this paper, we develop an efficient WRF Kessler microphysics scheme which runs on Graphics Processing Units (CPUs) using the NVIDIA Compute Unified Device Architecture (CUDA). The GPU-based implementation of Kessler microphysics scheme achieves a significant speedup of 70 × over its CPU based single-threaded counterpart. When a 4 GPU system is used, we achieve an overall speedup of 132 × as compared to the single thread CPU version.