Optimized thread-block arrangement in a GPU implementation of a linear solver for atmospheric chemistry mechanisms

Christian Guzman Ruiz,Mario C. Acosta,Oriol Jorba,Eduardo Cesar Galobardes,Matthew Dawson,Guillermo Oyarzun,Carlos Pérez García-Pando,Kim Serradell
DOI: https://doi.org/10.1016/j.cpc.2024.109240
IF: 4.717
2024-05-14
Computer Physics Communications
Abstract:Earth system models (ESM) demand significant hardware resources and energy consumption to solve atmospheric chemistry processes. Recent studies have shown improved performance from running these models on GPU accelerators. Nonetheless, there is room for improvement in exploiting even more GPU resources. This study proposes an optimized distribution of the chemical solver's computational load on the GPU, named Block-cells. Additionally, we evaluate different configurations for distributing the computational load in an NVIDIA GPU. We use the linear solver from the Chemistry Across Multiple Phases (CAMP) framework as our test bed. An intermediate-complexity chemical mechanism under typical atmospheric conditions is used. Results demonstrate a 35× speedup compared to the single-CPU thread reference case. Even using the full resources of the node (40 physical cores) on the reference case, the Block-cells version outperforms them by 50%. The Block-cells approach shows promise in alleviating the computational burden of chemical solvers on GPU architectures.
physics, mathematical,computer science, interdisciplinary applications
What problem does this paper attempt to address?