GPU acceleration of local and semilocal density functional calculations in the SPARC electronic structure code

Abstract

We present a GPU-accelerated version of the real-space SPARC electronic structure code for performing Kohn-Sham density functional theory calculations within the local density and generalized gradient approximations. In particular, we develop a modular math kernel based implementation for NVIDIA architectures wherein the computationally expensive operations are carried out on the GPUs, with the remainder of the workload retained on the CPUs. Using representative bulk and slab examples, we show that GPUs enable speedups of up to 6x relative to CPU-only execution, bringing time to solution down to less than 30 seconds for a metallic system with over 14,000 electrons, and enabling significant reductions in computational resources required for a given wall time.

arXiv comment11 pages, 3 figures

References in corpus (20)

• Quantum ESPRESSO: a modular and open-source software project for quantum simulations of materials
• Optimized norm-conserving Vanderbilt pseudopotentials
• Perspective on density functional theory
• O(N) methods in electronic structure calculations
• Real-Space Mesh Techniques in Density Functional Theory
• Daubechies wavelets as a basis set for density functional pseudopotential calculations
• Multiresolution analysis of electronic structure: semicardinal and wavelet bases
• Adaptive local basis set for Kohn-Sham density functional theory in a discontinuous Galerkin framework I: Total energy calculation
• Recent progress with large-scale ab initio calculations: the CONQUEST code
• Parallel Self-Consistent-Field Calculations via Chebyshev-Filtered Subspace Acceleration
• SPARC: Accurate and efficient finite-difference formulation and parallel implementation of Density Functional Theory: Extended systems
• New Algebraic Formulation of Density Functional Calculation
• SPARC: Accurate and efficient finite-difference formulation and parallel implementation of Density Functional Theory: Isolated clusters
• Periodic Pulay method for robust and efficient convergence acceleration of self-consistent field iterations
• SQDFT: Spectral Quadrature method for large-scale parallel $\mathcal{O}(N)$ Kohn-Sham calculations at high temperature
• GPGPU Acceleration of All-Electron Electronic Structure Theory Using Localized Numeric Atom-Centered Basis Functions
• Symmetry-adapted real-space density functional theory for cylindrical geometries: application to large X (X=C, Si, Ge, Sn) nanotubes
• On preconditioning the self-consistent field iteration in real-space Density Functional Theory
• On the calculation of the stress tensor in real-space Kohn-Sham Density Functional Theory
• Ab-initio investigation of finite size effects in rutile titania nanoparticles with semilocal and nonlocal density functionals

Cited by in corpus (1)

• Roadmap on Electronic Structure Codes in the Exascale Era