Air quality models are the silent workhorses behind every pollution warning that reaches a city’s residents, but their appetite for computing power has long been a brake on how finely they can resolve the atmosphere. A team of Chinese researchers has now reported a dramatic way forward. In a study published in Geoscientific Model Development, Kai Cao of the Institute of Atmospheric Physics at the Chinese Academy of Sciences, Qizhong Wu of Beijing Normal University, Xiao Tang of the Institute of Atmospheric Physics, and their colleagues describe how they rebuilt the advection module of the EPICC-Model, an open-source air quality modeling system developed for China’s complex pollution environment, so that it runs on graphics processing units. The result is a solver that, in offline benchmarks, executes up to 556.5 times faster than the original Fortran code running on a central processing unit, and a coupled model that completes realistic month-long simulations dramatically faster than before.
The bottleneck the team attacked is one that anyone who has pushed a weather or air quality model to higher resolution knows well. Doubling the horizontal resolution of an air quality model quadruples the number of grid cells, and numerical stability then forces the model to take proportionally shorter time steps. The combined effect is punishing: previous research cited in the study estimates that increasing horizontal resolution eighteenfold inflates the computational load three hundred times. Operational forecasting systems today mostly run at horizontal resolutions of several to tens of kilometers, which is too coarse for the urban-scale, street-by-street predictions that city managers increasingly demand. Faster hardware and smarter code are therefore not luxuries but prerequisites for the next generation of forecasts.
Graphics processing units offer a way out, because their thousands of parallel processors excel at exactly the kind of repetitive grid arithmetic that atmospheric models perform. The scale of modern GPU power is striking: the El Capitan supercomputer, which topped the TOP500 list in late 2024, delivers 1.742 exaflops of double-precision performance largely through AMD Instinct MI300A accelerators, each of which alone outperforms the entire Earth-Simulator, the world’s fastest machine in 2003. Yet most atmospheric models, including EPICC-Model, were written in Fortran for CPUs, with loop structures and data flows that translate poorly to GPUs. Simply copying code across rarely works; the algorithms must be rethought for the hardware.
The EPICC-Model team had already seen this dynamic play out within their own code. In earlier work, Cao and colleagues ported the model’s gas-phase chemistry module, historically the dominant cost at more than 40 percent of simulation time, onto China’s domestically developed GPU-like accelerators using the Heterogeneous-compute Interface for Portability, or HIP, a programming framework analogous to NVIDIA’s CUDA. That port cut the chemistry module’s share of runtime from 45.7 percent to 29.9 percent, but it exposed a new culprit: the horizontal advection module, which moves pollutants across the model grid with the wind, jumped from 13.3 percent to 25.0 percent of total runtime and became the model’s leading bottleneck. The new study targets exactly that module.
The heart of the advection scheme is the piecewise parabolic method, or PPM, a high-order Godunov-type algorithm from 1984 that approximates transported quantities with parabolic functions and is prized for its precision and robustness. It is the same scheme used in the latest versions of the widely used CMAQ and CAMx models. The researchers first rewrote the Fortran HADVPPM subroutine in standard C, then extended it with HIP calls so it could run on the GPU-like accelerator, which has 3840 computing units and 16 gigabytes of device memory, at the Earth System Numerical Simulation Facility (EarthLab) in Beijing. The resulting package, GPU-HADVPPM4HIP V1.0, follows the familiar heterogeneous computing sequence: allocate device memory, copy data from CPU to accelerator, launch parallel kernels, and copy results back.
Precision was checked at every step. Comparing the original Fortran, the rewritten C, and the HIP versions on identical inputs, the team found mean relative errors on the order of 10 to the minus 7 percent under default compilation, rising only slightly to around 10 to the minus 6 percent with aggressive O3 optimization. Interestingly, most of the small error crept in during the Fortran-to-C rewrite rather than the C-to-HIP step, which makes sense because HIP is essentially an extension of standard C. In a full 28-hour coupled simulation, the GPU version reproduced the Fortran model’s fields almost exactly, with differences in ozone of about 10 to the minus 2 ppbV and in PM2.5 of about 10 to the minus 3 micrograms per cubic meter, and normalized mean biases near zero.
The performance story unfolded in stages. A naive coupling, dubbed HIP-Ori, was actually a disaster: because the model called the advection solver roughly 9.8 million times per time step, each call triggering its own CPU-to-GPU data transfers, a one-hour integration ballooned from 1015 seconds in the original Fortran code to 20,400 seconds. The first fix, HIP-Opt1, restructured the loop hierarchy and expanded one-dimensional arrays into four- and five-dimensional blocks so that entire pollutant categories could be processed in single GPU calls, cutting transfers from about 9.8 million to just 10 per time step. That single change yielded a 17-fold improvement in model-level efficiency. A second optimization, HIP-Opt2, assigned each horizontal grid cell to its own GPU thread through coordinated thread-block indexing, after decoupling iterative calculations so that no step depended on the previous one, adding another 1.5-fold gain.
The team also implemented a hybrid MPI-plus-HIP scheme across EarthLab’s heterogeneous cluster, mapping each CPU process to a dedicated GPU accelerator so that the model scales across many nodes. In the final configuration, a 28-hour simulation that took the unoptimized GPU version 149.7 hours of wall-clock time finished in just 6.0 hours, roughly a 25-fold cumulative improvement over the naive port. At the module level, the GPU advection solver including data transfer overhead ran 39.3 percent faster than the Fortran original; with transfers excluded, the kernel itself was 20.5 times faster. Offline, on an array of 100 million double-precision values, the HIP solver compiled with O3 optimization reached that headline 556.5-fold speedup over the unoptimized Fortran baseline, and still managed 128-fold when both were compiled with the same aggressive flags.
Perhaps the most instructive finding is where the remaining time goes. Profiling showed that for large inputs, memory copying between CPU and GPU consumes roughly half of total execution time, a share that keeps growing with data size. This confirms that data movement, not arithmetic, is the defining constraint of heterogeneous computing, and it explains why the coupled model gains far less than the raw kernel speedup. The authors point to remedies including pinned-memory chunked transfers, asynchronous communication, unified memory architectures, and mixed-precision computation, as well as newer PCIe interconnects whose per-lane data rates have climbed from 16 gigatransfers per second in PCIe 4.0, used at EarthLab, to a projected 128 in PCIe 7.0.
The work also carries a broader message about scientific software in the GPU era. The EPICC-Model with its new advection solver reproduced observed PM2.5 and ozone variations reasonably well across eight Chinese cities during July 2021, with mean correlation coefficients of 0.70 for PM2.5 and 0.76 for ozone, demonstrating that the accelerated code remains scientifically trustworthy on domestic hardware. As the authors note, achieving kilometer-scale pollution simulations, of the kind that recently won a Gordon Bell Prize for climate modeling using NVIDIA GH200 GPUs, will require adapting entire air quality models to heterogeneous supercomputers, where nine of the world’s ten fastest machines now reside. Porting a single module is only the beginning; the real challenge, and opportunity, lies in redesigning the whole computational workflow around the GPU’s massively parallel architecture.
Subject of Research: GPU acceleration of the advection module in the EPICC-Model air quality modeling system
Article Title: Enhancing the advection module performance in the EPICC-Model V1.6.0 via GPU-HADVPPM4HIP V1.0 coupling and GPU-optimized strategies
Article References: Cao, K., Wu, Q., Tang, X., Li, J., Chen, X., Chen, H., Wang, W., Wu, H., Kong, L., Li, J., Zhu, J., & Wang, Z. (2026). Enhancing the advection module performance in the EPICC-Model V1.6.0 via GPU-HADVPPM4HIP V1.0 coupling and GPU-optimized strategies. Geoscientific Model Development, 19(19), 9807-9829. https://doi.org/10.5194/gmd-19-9807-2026
Image Credits: AI Generated
Keywords: GPU computing, air quality modeling, EPICC-Model, advection, piecewise parabolic method, HIP, high-performance computing, heterogeneous computing, EPICC-Model V1.6.0, GPU-HADVPPM4HIP, EarthLab, numerical weather prediction
News Source: Russell Cooper. (October 13, 2026). GPU Acceleration Supercharges Air Quality Model’s Advection Engine by Over 500 Times. Scienmag.
