Gpu speeds up cardiac interface model simulations on large 3d geometries

Multi-GPU Acceleration of Interface Problems in FEniCS, with Application to Cardiac Modeling

Distributed, Parallel, and Cluster ComputingMathematical Software

Summary

Simulating how heart cells interact through their membranes is important but can be very slow on traditional computers. The authors improved a popular simulation tool to run these complex models on graphics cards, which can do many tasks at once. Their approach lets the whole process, from building the math problem to solving it, run faster on GPUs, especially for big heart models. They tested it on both simple and realistic heart shapes, showing notable speed gains compared to powerful CPUs. This work helps speed up detailed heart simulations and could be applied to other problems involving interfaces.

What this means in practice

  • For computational cardiologists: Run large-scale heart cell interface simulations faster using GPUs to better study cardiac electrical behavior.
  • For computational engineers: Accelerate finite element problems involving interfaces in complex geometries by using the extended GPU-enabled FEniCS framework.

Authors

Edoardo Centofanti, Benjamin Pachev, Joshua Steyer, James D. Trotter, Pietro Benedusi

Abstract

GPU acceleration of finite element assembly is well established for standard formulations, but mixed-dimensional problems, in which volumetric subdomains are coupled via lower-dimensional interfaces, remain unsupported in automated frameworks such as FEniCS. We extend cuDOLFINx, a GPU-enabled extension of the DOLFINx finite element framework, to assemble such interface problems on GPUs while preserving the high-level FEniCS programming model. Combined with GPU-resident Krylov solvers and algebraic multigrid preconditioning, this enables end-to-end GPU execution of the EMI (Extracellular--Membrane--Intracellular) model, an elliptic interface problem from cardiac modeling in which many intracellular subdomains are coupled through membranes. We compare the GPU implementation against a CPU-based one through strong and weak scaling experiments on regular and realistic 3D cardiac geometries with up to 8000 cells and 22 million degrees of freedom. In terms of assembly, a single GH200 node is roughly equivalent to two 144-core CPU nodes in strong scaling, and to two to three nodes in weak scaling, depending on geometry. For the linear solver, performance depends on geometry, preconditioning, and problem scale. On the regular geometry, GPU advantages emerge mainly at larger problem sizes and node counts. On the realistic geometry, with the same algebraic multigrid preconditioner for both architectures, the GPU is faster at all tested scales. The implementation is available as open-source software and applies to other problems with interface coupling.