A modular, composite framework for the utilization of reduced-scaling Coulomb and exchange construction algorithms: Design and implementation

Abstract

Multiple algorithms exist for calculating Coulomb (J) or exchange (K) contributions to Fock-like matrices, and it is beneficial to develop a framework that allows the seamless integration and combination of different J and K construction algorithms. In Psi4, we have implemented the “CompositeJK” formalism for this purpose. CompositeJK allows for the combination of any J and K construction algorithms for any quantum chemistry method formulated in terms of J-like or K-like matrices (including, but not limited to, Hartree–Fock and density functional theory) in a highly modular and intuitive fashion, which is simple to utilize for both developers and users. Using the CompositeJK framework, Psi4 was interfaced to the sn-LinK implementation in the GauXC library, adding the first instance of noncommercial graphics processing unit (GPU) support for the construction of Fock matrix elements to Psi4. On systems with hundreds of atoms, the interface to the CPU sn-LinK implementation displays a higher performance than all the alternative JK construction methods available in Psi4, with up to x2.8 speedups compared to existing Psi4JK implementations. The GPU sn-LinK implementation, harnessing the power of GPUs, improves the observed performance gains to up to x7.0.