$\mathrm{DRAGON}$: A multi-GPU orbital-free density functional theory molecular dynamics simulation package for modeling of warm dense matter

Abstract

As progress in electronic structure theoretical methods is made, ab initio molecular dynamics (MD) based on orbital-free density functional theory (OF-DFT) is becoming increasingly more successful at substituting the traditional, very accurate but computationally costly Kohn–Sham (KS) approach for simulations of matter at the challenging warm dense matter (WDM) regime. However, despite the significant cost alleviation of eliminating the dependence on the KS orbitals, OF-DFT MD runs require ~10 2 to 10 3 CPU cores running for days, or even weeks, for simulations of systems comprised of 10 2 to 10 3 atoms, depending on thermodynamic conditions. Here we present DRAGON, a multi-GPU OF-DFT MD code for fast and efficient simulations of WDM. With a relatively small allocation of resources (4 to 8 GPU devices) it can provide an order of magnitude speedup for simulations containing $\mathscr{O}$(10 4 ) atoms and target systems composed of $\mathscr{O}$(10 5 ) atoms at conditions within the WDM regime, which is currently outside the capabilities of CPU codes.