Towards exascale fully relativistic pseudopotential density functional theory calculations enabled by mixed-precision computation and compressed-communication using residual based subspace iteration

Abstract

Materials exhibiting noncollinear magnetism or strong spin-orbit-coupling underpin many spintronic and topological applications, but their simulations require complex two-component spinors and costs substantially more than scalar density functional theory (DFT). We present a GPU-centric exascale finite-element DFT framework with noncollinear magnetism and spin-orbit-coupling, combining (i) adaptive higher-order finite-element discretization, (ii) a matrix-free Poisson solver, (iii) residual-based Chebyshev filtered subspace iteration (R-ChFSI) for sparse generalized eigenproblems, (iv) R-ChFSI-enabled mixed precision computation with block floating-point compressed MPI communication, and (v) communication-efficient band partitioning. R-ChFSI permits inexact matrix-multivector products and $4\times$--$6.4\times$ compression of communicated data relative to FP64 while preserving double-precision robustness and reducing computation and data movement. Results demonstrate strong scaling on Aurora and Frontier, with up to $3.4\times$ faster Chebyshev filtering. On Aurora, total self-consistent-field (SCF) solve wall time reduces by $2\times$. An 800-node (9,600-GPU) calculation demonstrates fully relativistic pseudopotential DFT for $\sim$80,000 electrons in under 7 minutes per SCF iteration.