GPU-Accelerated Nested Schur-Complement Solvers for Atmospheric Flow Simulations Using Horizontally-Explicit, Vertically-Implicit Schemes and Discontinuous Galerkin Methods
Abstract
Horizontally explicit, vertically implicit (HEVI) time integration removes the acoustic time-step restriction imposed by fine vertical grids in nonhydrostatic atmospheric models, but requires fast solvers for the resulting vertical linear systems, especially on GPUs. We present GPU-accelerated nested Schur-complement solvers for these systems for a split-form discontinuous Galerkin spectral element method on Legendre-Gauss-Lobatto (LGL) nodes applied to the compressible Euler equations with gravity. In time, we use a Rosenbrock-W method, a linearly implicit Runge-Kutta scheme that retains its order and stability with approximate Jacobians. This allows a sparse vertical Jacobian that captures vertical acoustic waves and gravity on curvilinear meshes with orography and decouples into independent column systems. The solvers are nested: a global Schur complement (static condensation) exploits the LGL sparsity to condense each column onto a few element-edge unknowns, and a local Schur complement eliminates all variables but the vertical momentum within each element, exploiting that the pressure depends only on the potential-temperature density. We propose and analyze two condensation strategies, which lead to banded and block-tridiagonal reduced systems. The resulting HEVI schemes conserve total mass and potential-temperature density to round-off. The solvers are implemented for CPUs and GPUs in TrixiAtmo.jl, the atmospheric extension of Trixi.jl. We verify high-order convergence with linear inertia-gravity waves in a 2D channel and demonstrate robustness on the sphere with the baroclinic instability, a mountain-induced baroclinic instability, and the DCMIP-2025 gap-flow and vortex-shedding cases. For the baroclinic instability without orography, the scheme achieves 18.8 and 2.9 simulated years per day at 100 and 50 km resolution on one NVIDIA GH200 GPU in double precision.
Related
- Data driven non-equilibrium moist phase exchanges for atmospheric convection within a discontinuous Galerkin model of the compressible Euler equations
- A tool for new Hybrid models: a swift algorithm to calculate the left inverse of a square root of a horizontally localized climatological ensemble covariance matrix
- Discrete hypocoercive estimates for discontinuous Galerkin methods: application to the Vlasov-Poisson-Fokker-Planck system
- The discontinuous Galerkin method for solving the two-dimensional transport equation with distributed-order memory kernel on triangular meshes
- Heat equation with singular drift which has bounded solution discontinuous at an interior point
BibTeX
@misc{artiano2026gpuaccelerated,
title = {{GPU-Accelerated Nested Schur-Complement Solvers for Atmospheric Flow Simulations Using Horizontally-Explicit, Vertically-Implicit Schemes and Discontinuous Galerkin Methods}},
author = {Marco Artiano and Andrés Rueda-Ramírez and Oswald Knoth and Hendrik Ranocha and Gregor Gassner},
howpublished = {arXiv},
year = {2026},
month = oct,
doi = {10.48550/arxiv.2610.15775},
eprint = {2610.15775},
archivePrefix = {arXiv},
url = {https://arxiv.org/abs/2610.15775},
note = {CC BY 4.0},
}
RIS
TY - UNPB
TI - GPU-Accelerated Nested Schur-Complement Solvers for Atmospheric Flow Simulations Using Horizontally-Explicit, Vertically-Implicit Schemes and Discontinuous Galerkin Methods
AU - Marco Artiano
AU - Andrés Rueda-Ramírez
AU - Oswald Knoth
AU - Hendrik Ranocha
AU - Gregor Gassner
PY - 2026
DA - 2026/10/11
PB - arXiv
DO - 10.48550/arxiv.2610.15775
UR - https://arxiv.org/abs/2610.15775
AB - Horizontally explicit, vertically implicit (HEVI) time integration removes the acoustic time-step restriction imposed by fine vertical grids in nonhydrostatic atmospheric models, but requires fast solvers for the resulting vertical linear systems, especially on GPUs. We present GPU-accelerated nested Schur-complement solvers for these systems for a split-form discontinuous Galerkin spectral element method on Legendre-Gauss-Lobatto (LGL) nodes applied to the compressible Euler equations with gravity. In time, we use a Rosenbrock-W method, a linearly implicit Runge-Kutta scheme that retains its order and stability with approximate Jacobians. This allows a sparse vertical Jacobian that captures vertical acoustic waves and gravity on curvilinear meshes with orography and decouples into independent column systems. The solvers are nested: a global Schur complement (static condensation) exploits the LGL sparsity to condense each column onto a few element-edge unknowns, and a local Schur complement eliminates all variables but the vertical momentum within each element, exploiting that the pressure depends only on the potential-temperature density. We propose and analyze two condensation strategies, which lead to banded and block-tridiagonal reduced systems. The resulting HEVI schemes conserve total mass and potential-temperature density to round-off. The solvers are implemented for CPUs and GPUs in TrixiAtmo.jl, the atmospheric extension of Trixi.jl. We verify high-order convergence with linear inertia-gravity waves in a 2D channel and demonstrate robustness on the sphere with the baroclinic instability, a mountain-induced baroclinic instability, and the DCMIP-2025 gap-flow and vortex-shedding cases. For the baroclinic instability without orography, the scheme achieves 18.8 and 2.9 simulated years per day at 100 and 50 km resolution on one NVIDIA GH200 GPU in double precision.
ER -