A fast, dense Chebyshev solver for electronic structure on GPUs

Joshua Finkelstein; Christian F A Negre; Jean-Luc Fattebert

doi:10.1063/5.0164255

A fast, dense Chebyshev solver for electronic structure on GPUs

J Chem Phys. 2023 Sep 14;159(10):101101. doi: 10.1063/5.0164255.

Authors

Joshua Finkelstein¹, Christian F A Negre¹, Jean-Luc Fattebert²

Affiliations

¹ Theoretical Division, Los Alamos National Laboratory, Los Alamos, New Mexico 87545, USA.
² Computational Sciences and Engineering Division, Oak Ridge National Laboratory, Oak Ridge, Tennessee 37830, USA.

PMID: 37694745
DOI: 10.1063/5.0164255

Abstract

Matrix diagonalization is almost always involved in computing the density matrix needed in quantum chemistry calculations. In the case of modest matrix sizes (≲4000), performance of traditional dense diagonalization algorithms on modern GPUs is underwhelming compared to the peak performance of these devices. This motivates the exploration of alternative algorithms better suited to these types of architectures. We newly derive, and present in detail, an existing Chebyshev expansion algorithm [Liang et al., J. Chem. Phys. 119, 4117-4125 (2003)] whose number of required matrix multiplications scales with the square root of the number of terms in the expansion. Focusing on dense matrices of modest size, our implementation on GPUs results in large speed ups when compared to diagonalization. Additionally, we improve upon this existing method by capitalizing on the inherent task parallelism and concurrency in the algorithm. This improvement is implemented on GPUs by using CUDA and HIP streams via the MAGMA library and leads to a significant speed up over the serial-only approach for smaller (≲1000) matrix sizes. Finally, we apply our technique to a model system with a high density of states around the Fermi level, which typically presents significant challenges.