Solver composition across the PDE/linear algebra barrier

Abstract

The efficient solution of discretisations of coupled systems of partial differential equations (PDEs) is at the core of much of numerical simulation. Significant effort has been expended on scalable algorithms to precondition Krylov iterations for the linear systems that arise. With few exceptions, the reported numerical implementation of such solution strategies is specific to a particular model setup, and intimately ties the solver strategy to the discretisation and PDE, especially when the preconditioner requires auxiliary operators. In this paper, we present recent improvements in the Firedrake finite element library that allow for straightforward development of the building blocks of extensible, composable preconditioners that decouple the solver from the model formulation. Our implementation extends the algebraic composability of linear solvers offered by the PETSc library by augmenting operators, and hence preconditioners, with the ability to provide any necessary auxiliary operators. Rather than specifying up front the full solver configuration, tied to the model, solvers can be developed independently of model formulation and configured at runtime. We illustrate with examples from incompressible fluids and temperature-driven convection.

23 pages. Fixed axis labelling in Fig 3

Cited by in corpus (13)

• An augmented Lagrangian preconditioner for the 3D stationary incompressible Navier-Stokes equations at high Reynolds number
• Higher-order compatible finite element schemes for the nonlinear rotating shallow water equations on the sphere
• A block preconditioner for non-isothermal flow in porous media
• Code generation for generally mapped finite elements
• Phase-field modeling of multivariant martensitic transformation at finite-strain: computational aspects and large-scale finite-element simulations
• A study of vectorization for matrix-free finite element methods
• Slate: extending Firedrake's domain-specific abstraction to hybridized solvers for geoscience and beyond
• Monolithic multigrid for implicit Runge-Kutta discretizations of incompressible fluid flow
• Low-order preconditioning of the Stokes equations
• A scalable DG solver for the electroneutral Nernst-Planck equations
• Finite element methods for large-strain poroelasticity/chemotaxis models simulating the formation of myocardial oedema
• Robust finite element methods and solvers for the Biot--Brinkman equations in vorticity form
• asQ: parallel-in-time finite element simulations using ParaDiag for geoscientific models and beyond