GPGPU Acceleration of All-Electron Electronic Structure Theory Using Localized Numeric Atom-Centered Basis Functions

Abstract

We present an implementation of all-electron density-functional theory for massively parallel GPGPU-based platforms, using localized atom-centered basis functions and real-space integration grids. Special attention is paid to domain decomposition of the problem on non-uniform grids, which enables compute- and memory-parallel execution across thousands of nodes for real-space operations, e.g. the update of the electron density, the integration of the real-space Hamiltonian matrix, and calculation of Pulay forces. To assess the performance of our GPGPU implementation, we performed benchmarks on three different architectures using a 103-material test set. We find that operations which rely on dense serial linear algebra show dramatic speedups from GPGPU acceleration: in particular, SCF iterations including force and stress calculations exhibit speedups ranging from 4.5 to 6.6. For the architectures and problem types investigated here, this translates to an expected overall speedup between 3-4 for the entire calculation (including non-GPU accelerated parts), for problems featuring several tens to hundreds of atoms. Additional calculations for a 375-atom Bi${}_2$Se${}_3$ bilayer show that the present GPGPU strategy scales for large-scale distributed-parallel simulations.

arXiv comment49 pages, 9 figures

References in corpus (14)

• The SIESTA method for ab initio order-N materials simulation
• Efficient index handling of multidimensional periodic boundary conditions
• Perspective on density functional theory
• Resolution-of-identity approach to Hartree-Fock, hybrid density functionals, RPA, MP2, and \textit{GW} with numeric atom-centered orbital basis functions
• O(N) methods in electronic structure calculations
• VASP on a GPU: application to exact-exchange calculations of the stability of elemental boron
• ELSI: A Unified Software Interface for Kohn-Sham Electronic Structure Solvers
• The Elephant in the Room of Density Functional Theory Calculations
• Accelerating Atomic Orbital-based Electronic Structure Calculation via Pole Expansion and Selected Inversion
• Real-space density functional theory on graphical processing units: computational approach and comparison to Gaussian basis set methods
• GPU-Acceleration of the ELPA2 Distributed Eigensolver for Dense Symmetric and Hermitian Eigenproblems
• Thermodynamic equilibrium conditions of graphene films on SiC
• Optimizations of the Eigensolvers in the ELPA Library
• Graphics Processing Unit acceleration of the Random Phase Approximation in the projector augmented wave method

Cited by in corpus (17)

• Roadmap on Electronic Structure Codes in the Exascale Era
• GPU Acceleration of Large-Scale Full-Frequency GW Calculations
• ELSI -- An Open Infrastructure for Electronic Structure Solvers
• GPU-Acceleration of the ELPA2 Distributed Eigensolver for Dense Symmetric and Hermitian Eigenproblems
• Large scale quantum chemistry with Tensor Processing Units
• Distributed Memory, GPU Accelerated Fock Construction for Hybrid, Gaussian Basis Density Functional Theory
• Mixed Precision Fermi-Operator Expansion on Tensor Cores From a Machine Learning Perspective
• GPU acceleration of local and semilocal density functional calculations in the SPARC electronic structure code
• Top-down fabrication of atomic patterns in twisted bilayer graphene
• GPU-Accelerated Solution of the Bethe-Salpeter Equation for Large and Heterogeneous Systems
• Accurate Frozen Core Approximation for All-Electron Density-Functional Theory
• An alternative GPU acceleration for a pseudopotential plane-waves density functional theory code with applications to metallic systems
• Structural Anisotropy in Sb Thin Films
• INQ, a modern GPU-accelerated computational framework for (time-dependent) density functional theory
• On the Efficient Evaluation of the Exchange Correlation Potential on Graphics Processing Unit Clusters
• Eat, Sleep, Code, Repeat: Tips for Early-Career Researchers in Computational Science
• GPU Acceleration of Collinear and Noncollinear DFT Using a Numerical Atomic Orbital-Based DFT Code