Porting Fragmentation Methods to Graphical Processing Units Using an OpenMP Application Programming Interface: Offloading the Fock Build for Low Angular Momentum Functions

A framework to offload four-index two-electron repulsion integrals to graphical processing units (GPUs) using OpenMP is discussed. The method has been applied to the Fock build for low angular momentum s and p functions in both the restricted Hartree-Fock (RHF) and in the effective fragment molecular orbital (EFMO) framework. Benchmark calculations for the GPU code for the pure RHF method show an increasing speedup relative to the existing OpenMP CPU code in GAMESS from 1.04 to 52x for clusters of 70-569 water molecules. The parallel efficiency on 24 NVIDIA V100 GPU boards also increases when increasing the system size: from 75 to 94% for water clusters that contain 303-1120 molecules. In the EFMO framework, the GPU Fock build shows a high linear scalability up to 4608 V100s with a parallel efficiency of 96% for calculations on a solvated mesoporous silica nanoparticle system with similar to 67,000 basis functions.

Automated summary

This paper discusses a framework for offloading four-index two-electron repulsion integrals to GPUs using OpenMP. The framework is applied to Fock build for low angular momentum s and p functions in both RHF and EFMO. Benchmark calculations show that the GPU code outperforms the existing CPU code in GAMESS, with a speedup ranging from 1.04 to 52x for clusters of water molecules. In addition, the parallel efficiency of the GPU code increases with system size in both RHF and EFMO.