Subspace recursive Fermi-operator expansion strategies for large-scale DFT eigenvalue problems on HPC architectures
arXiv:2301.04642 · doi:10.1063/5.0150287
Abstract
Quantum mechanical calculations for material modelling using Kohn-Sham density functional theory (DFT) involve the solution of a nonlinear eigenvalue problem for smallest eigenvector-eigenvalue pairs with proportional to the number of electrons in the material system. These calculations are computationally demanding and have asymptotic cubic scaling complexity with the number of electrons. Large-scale matrix eigenvalue problems arising from the discretization of the Kohn-Sham DFT equations employing a systematically convergent basis traditionally rely on iterative orthogonal projection methods, which are shown to be computationally efficient and scalable on massively parallel computing architectures. However, as the size of the material system increases, these methods are known to incur dominant computational costs through the Rayleigh-Ritz projection step of the discretized Kohn-Sham Hamiltonian matrix and the subsequent subspace diagonalization of the projected matrix. This work explores the potential of polynomial expansion approaches based on recursive Fermi-operator expansion as an alternative to the subspace diagonalization of the projected Hamiltonian matrix to reduce the computational cost. Subsequently, we perform a detailed comparison of various recursive polynomial expansion approaches to the traditional approach of explicit diagonalization on both multi-node CPU and GPU architectures and assess their relative performance in terms of accuracy, computational efficiency, scaling behaviour and energy efficiency.
10 pages, 7 figures
References in corpus (5)
- The Kernel Polynomial Method
- DFT-FE 1.0: A massively parallel hybrid CPU-GPU density functional theory code using finite-element discretization
- Large scale quantum chemistry with Tensor Processing Units
- Non-monotonic recursive polynomial expansions for linear scaling calculation of the density matrix
- Mixed Precision Fermi-Operator Expansion on Tensor Cores From a Machine Learning Perspective