GPU Acceleration of Large-Scale Full-Frequency GW Calculations

Abstract

Many-body perturbation theory is a powerful method to simulate electronic excitations in molecules and materials starting from the output of density functional theory calculations. By implementing the theory efficiently so as to run at scale on the latest leadership high-performance computing systems it is possible to extend the scope of GW calculations. We present a GPU acceleration study of the full-frequency GW method as implemented in the WEST code. Excellent performance is achieved through the use of (i) optimized GPU libraries, e.g., cuFFT and cuBLAS, (ii) a hierarchical parallelization strategy that minimizes CPU-CPU, CPU-GPU, and GPU-GPU data transfer operations, (iii) nonblocking MPI communications that overlap with GPU computations, and (iv) mixed-precision in selected portions of the code. A series of performance benchmarks have been carried out on leadership high-performance computing systems, showing a substantial speedup of the GPU-accelerated version of WEST with respect to its CPU version. Good strong and weak scaling is demonstrated using up to 25920 GPUs. Finally, we showcase the capability of the GPU version of WEST for large-scale, full-frequency GW calculations of realistic systems, e.g., a nanostructure, an interface, and a defect, comprising up to 10368 valence electrons.

References in corpus (52)

• Quantum ESPRESSO: a modular and open-source software project for quantum simulations of materials
• Phonons and related properties of extended systems from density-functional perturbation theory
• Advanced capabilities for materials modelling with Quantum ESPRESSO
• CP2K: An Electronic Structure and Molecular Dynamics Software Package -- Quickstep: Efficient and Accurate Electronic Structure Calculations
• Optimized norm-conserving Vanderbilt pseudopotentials
• Quantum ESPRESSO toward the exascale
• The GW method
• Optimization Algorithm for the Generation of ONCV Pseudopotentials
• BerkeleyGW: A Massively Parallel Computer Package for the Calculation of the Quasiparticle and Optical Properties of Materials and Nanostructures
• NWChem: Past, Present, and Future
• Many-body perturbation theory calculations using the yambo code
• First-principles GW calculations for fullerenes, porphyrins, phtalocyanine, and other molecules of interest for organic photovoltaic applications
• The GW compendium: A practical guide to theoretical photoemission spectroscopy
• Octopus, a computational framework for exploring light-driven phenomena and quantum dynamics in extended and finite systems
• Polytype control of spin qubits in silicon carbide
• Implementation of an all-electron GW approximation based on the PAW method without plasmon pole approximation: application to Si, SiC, AlAs, InAs, NaH and KH
• Quasiparticle GW calculations for solids, molecules and 2D materials
• Optical excitations in organic molecules, clusters and defects studied by first-principles Green's function methods
• GW quasi-particle spectra from occupied states only
• Quantum simulations of materials on near-term quantum computers
• Cubic scaling $GW$: towards fast quasiparticle calculations
• Nonempirical Range-separated Hybrid Functionals for Solids and Molecules
• Band Offsets at the Si/SiO${}_2$ Interface from Many-Body Perturbation Theory
• Breaking the theoretical scaling limit for predicting quasi-particle energies: The stochastic GW approach
• Towards GW Calculations on Thousands of Atoms
• GW method with the self-consistent Sternheimer equation
• Quantum Embedding Theory for Strongly-correlated States in Materials
• An O(N^3) implementation of Hedin's GW approximation for molecules
• Cubic-scaling all-electron GW calculations with a separable density-fitting space-time approach
• Low-order Scaling $G_0{W}_0$ by Pair Atomic Density Fitting
• Photoluminescence spectra of point defects in semiconductors: validation of first principles calculations
• Swift $GW$ beyond $10,000$ electrons using fractured stochastic orbitals
• Low-scaling $GW$ with benchmark accuracy and application to phosphorene nanosheets
• Dielectric dependent hybrid functionals for heterogeneous materials
• Fundamental Principles for Calculating Charged Defect Ionization Energies in Ultrathin Two-Dimensional Materials
• Reproducibility in $G_0{W}_0$ Calculations for Solids
• Designing defect-based qubit candidates in wide-gap binary semiconductors for solid-state quantum technologies
• All-electron periodic $G_0{W}_0$ implementation with numerical atomic orbital basis functions: algorithm and benchmarks
• Enhanced Static Approximation to the Electron Self-Energy Operator for Efficient Calculation of Quasiparticle Energies
• Elimination of unoccupied state summations in ab-initio self-energy calculations for large supercells
• A finite field approach to solving the Bethe Salpeter equation
• GPU-Acceleration of the ELPA2 Distributed Eigensolver for Dense Symmetric and Hermitian Eigenproblems
• GPGPU Acceleration of All-Electron Electronic Structure Theory Using Localized Numeric Atom-Centered Basis Functions
• First-principles Studies of Strongly Correlated States in Defect Spin Qubits in Diamond
• Stochastic many-body perturbation theory for Moiré states in twisted bilayer phosphorene
• A Finite-field Approach for $GW$ Calculations Beyond the Random Phase Approximation
• Frequency dependence in GW made simple using a multi-pole approximation
• Quantum vibronic effects on the electronic properties of solid and molecular carbon
• Efficient dielectric matrix calculations using the Lanczos algorithm for fast many-body $G_0{W}_0$ implementations
• Faster Exact Exchange in Periodic Systems using Single-precision Arithmetic
• Scalable GW software for quasiparticle properties using OpenAtom
• Improving the efficiency of $G_0{W}_0$ calculations with approximate spectral decompositions of dielectric matrices

Cited by in corpus (2)

• Roadmap on Electronic Structure Codes in the Exascale Era
• Spatial Decay and Limits of Quantum Solute-Solvent Interactions