Tucker Tensor Approach for Accelerating Fock Exchange Computations in a Real-Space Finite-Element Discretization of Generalized Kohn-Sham Density Functional Theory

J Chem Theory Comput. 2024 May 14;20(9):3566-3579. doi: 10.1021/acs.jctc.4c00019. Epub 2024 Apr 25.

Abstract

The evaluation of Fock exchange is often the computationally most expensive part of hybrid functional density functional theory calculations in a systematically improvable, complete basis. In this work, we employ a Tucker tensor based approach that substantially accelerates the evaluation of the action of Fock exchange by transforming three-dimensional convolutional integrals into a tensor product of one-dimensional convolution integrals. Our numerical implementation uses a parallelization strategy that balances the memory and communication bottlenecks, alongside overlapping compute and communication operations to enhance computational efficiency and parallel scalability. The accuracy and computational efficiency are demonstrated on various systems, including Pt clusters of various sizes and a TiO2 cluster with 3684 electrons.