An alternative GPU acceleration for a pseudopotential plane-waves density functional theory code with applications to metallic systems

Abstract

We present an alternative GPU acceleration for plane waves pseudopotentials electronic structure codes designed for systems that have small unit cells but require a large number of k points to sample the Brillouin zone as happens, for instance, in metals. We discuss the diagonalization of the Kohn and Sham equations and the solution of the linear system derived in density functional perturbation theory. Both problems take advantage from a rewriting of the routine that applies the Hamiltonian to the Bloch wave-functions to work simultaneously (in parallel on the GPU threads) on the wave-functions with different wave-vectors k, as many as allowed by the GPU memory. Our implementation is written in CUDA Fortran and makes extensive use of kernel routines that run on the GPU (GLOBAL routines) or can be called from inside the GPU threads (DEVICE routines). We compare our method with the CPUs only calculation and with the approach currently implemented in Quantum ESPRESSO that uses GPU accelerated libraries for the FFT and for the linear algebra tasks such as the matrix-matrix multiplications as well as OpenACC directives for loop parallelization. We show in a realistic example that our method can give a significant improvement in the cases for which it has been designed.

References in corpus (13)

• Quantum ESPRESSO: a modular and open-source software project for quantum simulations of materials
• Restoring the density-gradient expansion for exchange in solids and surfaces
• Generalized gradient approximation for solids and their surfaces
• Phonons and related properties of extended systems from density-functional perturbation theory
• Advanced capabilities for materials modelling with Quantum ESPRESSO
• Quantum ESPRESSO toward the exascale
• Ab initio variational approach for evaluating lattice thermal conductivity
• Quasi-harmonic temperature dependent elastic constants: applications to silicon, aluminum, and silver
• GPGPU Acceleration of All-Electron Electronic Structure Theory Using Localized Numeric Atom-Centered Basis Functions
• Quasi-harmonic thermoelasticity of palladium, platinum, copper, and gold from first principles
• Ab initio quasi-harmonic thermoelasticity of molybdenum at high temperature and pressure
• Ab initio thermodynamic properties of Iridium: A high-pressure and high-temperature study
• High temperature and pressure thermoelasticity of hcp metals from ab initio quasi-harmonic free energy calculations: the beryllium case

Cited by in corpus (1)

• High pressure and temperature thermoelasticity of hcp osmium from ab initio quasi-harmonic theory