Adapting Atmospheric Chemistry Components for Efficient GPU Accelerators

Atmospheric models demand a lot of computational power, and solving the chemical processes is one of its most computationally intensive components. This work shows how to improve the computational performance of the Multiscale Online Nonhydrostatic AtmospheRe CHemistry (MONARCH), a chemical weather prediction system developed by the Barcelona Supercomputing Center. The model implements the new flexible external package chemistry across multiple phases (CAMP) for the solving of gas- and aerosol-phase chemical processes that allows multiple chemical processes to be solved simultaneously as a single system. We introduce a novel strategy to simultaneously solve multiple instances of a chemical mechanism, represented in the model as grid cells, obtaining a speedup up to 9 $$\times $$ using thousands of cells. In addition, we present a GPU strategy for the most time-consuming function of CAMP. The GPU version achieves up to 1.2 $$\times $$ speedup compared to CPU. Also, we optimize the memory access in the GPU to increase its speedup up to 1.7 $$\times $$ .