GPU 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 realspace 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.

Quantum-based Molecular Dynamics Simulations Using Tensor Cores

“…The development presented in this article mirrors a similar a transition that occurred over a decade ago when graphics processing units (GPU's) gradually became more accessible to scientific computing [23][24][25][26][27][28][29][30][31][32][33][34][35]. Some of the more demanding computational tasks performed on the general purpose central processing units (CPUs) were successfully transferred to the more specialized but higher performing GPUs.…”

“…[73] and our proposed algorithm in Eq. (32). To illustrate this method for a canonical (NVT) simulation, we use an 88 molecule water system in a periodic box.…”

ELSI: A unified software interface for Kohn–Sham electronic structure solvers

“…The computation power from GPUs has been utilized by the electronic structure community for larger and faster simulations [73, 74,75,76,77,78,79,80,81,82,83,84,85]. Various eigensolver and density matrix solver implementations targeting hybrid CPU-GPU machines have been reported.…”

Achieving performance portability in Gaussian basis set density functional theory on accelerator based architectures in NWChemEx

“…[4,5,23] for a review of modern trends as well as Refs. [24][25][26][27][28] for a number of recent developments).…”