Usage of Graphics Processing Units (GPU) has become strategic for simulating the chemistry of large molecular systems, with the majority of top supercomputers utilizing GPUs as their main source of computational horsepower. In this paper, a new fragmentation-based Hartree-Fock matrix build algorithm designed for scaling on many-GPU architectures is presented. The new algorithm uses a novel dynamic load balancing scheme based on a binned shell-pair container to distribute batches of significant shell quartets with the same code path to different GPUs. This maximizes computational throughput and load balancing, and eliminates GPU thread divergence due to integral screening. Additionally, the code uses a novel Fock digestion algorithm to contract electron repulsion integrals into the Fock matrix, which exploits all forms of permutational symmetry and eliminates thread synchronization requirements. The implementation demonstrates excellent scalability on the Summit computer, achieving good strong scaling performance up to 4096 nodes, and linear weak scaling up to 612 nodes.
|Title of host publication
|Proceedings of SC20
|Subtitle of host publication
|The International Conference for High Performance Computing, Networking, Storage and Analysis: Virtual Event, November 9-19, 2020
|Institute of Electrical and Electronics Engineers Inc.
|Number of pages
|Published - 2020
|2020 International Conference for High Performance Computing, Networking, Storage and Analysis - Virtual Conference, Atlanta, United States
Duration: 9 Nov 2020 → 19 Nov 2020
|2020 International Conference for High Performance Computing, Networking, Storage and Analysis
|9/11/20 → 19/11/20