arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2303.14280physics.comp-phphysics.chem-ph
收藏

分布式内存、GPU加速的混合高斯基组密度泛函理论Fock矩阵构建

Distributed Memory, GPU Accelerated Fock Construction for Hybrid, Gaussian Basis Density Functional Theory

David B. Williams-Young, Andrey Asadchev, Doru Thom Popovici, David Clark, Johnathan Waldrop, Theresa Windus, Edward F. Valeev, Wibe A. de Jong

AI总结:

提出一组用于混合Kohn-Sham DFT的分布式内存GPU加速算法,通过DF-J-Engine和sn-K方法评估Coulomb与精确交换矩阵,在Perlmutter上使用128个A100 GPU实现了数百至千原子规模的高效强扩展。

摘要:

随着现代超级计算机日益依赖GPU等加速器架构,开发并优化电子结构方法以利用这些大规模并行资源已成为近期的优先事项。尽管在多体(如耦合簇)和谱单体(如平面波、实空间和有限元密度泛函理论[DFT])方法的GPU加速分布式内存算法开发上已取得重大进展,但绝大多数GPU加速的高斯原子轨道方法仍集中于共享内存系统,仅有少数案例在分布式内存GPU架构上追求大规模并行。在本工作中,我们提出了一组分布式内存算法,分别通过直接密度拟合(DF-J-Engine)和半数值(sn-K)方法,评估具有高斯基组的混合Kohn-Sham DFT的Coulomb和精确交换矩阵。在Perlmutter超级计算机上使用多达128个NVIDIA A100 GPU,针对包含数百至一千多个原子的系统,展示了所开发方法的绝对性能和强可扩展性。

英文摘要:

With the growing reliance of modern supercomputers on accelerator-based architectures such a GPUs, the development and optimization of electronic structure methods to exploit these massively parallel resources has become a recent priority. While significant strides have been made in the development of GPU accelerated, distributed memory algorithms for many-body (e.g. coupled-cluster) and spectral single-body (e.g. planewave, real-space and finite-element density functional theory [DFT]), the vast majority of GPU-accelerated Gaussian atomic orbital methods have focused on shared memory systems with only a handful of examples pursuing massive parallelism on distributed memory GPU architectures. In the present work, we present a set of distributed memory algorithms for the evaluation of the Coulomb and exact-exchange matrices for hybrid Kohn-Sham DFT with Gaussian basis sets via direct density-fitted (DF-J-Engine) and seminumerical (sn-K) methods, respectively. The absolute performance and strong scalability of the developed methods are demonstrated on systems ranging from a few hundred to over one thousand atoms using up to 128 NVIDIA A100 GPUs on the Perlmutter supercomputer.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑