PASCI : A Scalable Framework for Heterogeneous Parallel Calculation of Dynamical Electron Correlation

Runfeng Jin,Wenhao Liang,Haoyuan Zhang,Yinxuan Song,Zhen Luo,Haibo Ma,Yingjin Ma,Zhong Jin
DOI: https://doi.org/10.1145/3673038.3673098
2024-01-01
Abstract:Accurately calculating the electronic structure of strongly correlated chemical systems necessitates a detailed description of both static and dynamical electron correlations, posing a significant challenge in ab initio quantum chemistry. Although the high memory and computational demands generally limit these calculations to relatively modest systems, the advanced computational capabilities of modern GPUs provide new avenues to expand these limits. However, complex control flows inherent to computation notably impair performance on GPUs. Furthermore, the significant disparity in computational load across different branches leads to load imbalance, challenging the large-scale simulations. In this work, we introduce PASCI, a heterogeneous parallel computing framework designed to quickly and efficiently parallelize the computation of dynamical correlation energy based on determinants. The features of the PASCI framework include (1) a divergence-avoiding GPU algorithm, (2) a three-level load-mapping strategy to ensure load balance across processors, GPU warps, and GPU threads, (3) performance models for memory footprint and computation, and (4) seamless integration with existing quantum chemistry software. Experimental results using an NVIDIA A100 GPU demonstrate that our new GPU algorithm achieves an average 6.6x (up to 13.8x) peak performance increase and 2-4 orders of magnitude speedup in practical usage compared to its original GPU implementation. Moreover, PASCI exhibits excellent scalability, highlighting its potential as a powerful high-performance computing tool in complex quantum chemistry research.
What problem does this paper attempt to address?