.png){kind=icon}
返回
Multinode Multi-GPU Two-Electron Integrals: Code Generation Using the Regent Language
.png){kind=icon}
PRE
AI
DOI：10.1021/acs.jctc.2c00414.png)
.png){kind=icon}
原文求助.png){kind=icon}
分享.png){kind=icon}
收藏
摘要
En 中文
The computation of two-electron repulsion integrals (ERIs) is often the most expensive step of integral-direct self-consistent field methods. Formally it scales as O(N-4), where N is the number of Gaussian basis functions used to represent the molecular wave function. In practice, this scaling can be reduced to O(N-2) or less by neglecting small integrals with screening methods. The contributions of the ERIs to the Fock matrix are of Coulomb (J) and exchange (K) type and require separate algorithms to compute matrix elements efficiently. We previously implemented highly efficient GPU-accelerated J-matrix and K-matrix algorithms in the electronic structure code TeraChem. Although these implementations supported the use of multiple GPUs on a node, they did not support the use of multiple nodes. This presents a key bottleneck to cutting-edge ab initio simulations of large systems, e.g., excited state dynamics of photoactive proteins. We present our implementation of multinode multi-GPU J- and K-matrix algorithms in TeraChem using the Regent programming language. Regent directly supports distributed computation in a task-based model and can generate code for a variety of architectures, including NVIDIA GPUs. We demonstrate multinode scaling up to 45 GPUs (3 nodes) and benchmark against hand-coded TeraChem integral code. We also outline our metaprogrammed Regent implementation, which enables flexible code generation for integrals of different angular momenta.
Keyword:
GRAPHICAL PROCESSING UNITS
QUANTUM-CHEMISTRY CALCULATIONS
MOLECULAR INTEGRALS
COUPLED-CLUSTER
EXCITED-STATES
CONFIGURATION-INTERACTION
COULOMB OPERATOR
HIGH-PERFORMANCE
FOCK-MATRIX
EFFICIENT
期刊
IF:
5.5
论文数:
1.1W
被引数:
5.4W
机构
引用论文
.png){kind=icon}
Massively Parallel Quantum Chemistry: A high-performance research platform for electronic structure
OAAI
OAAI分享收藏分享收藏Spin-restricted ensemble-referenced Kohn-Sham method: basic principles and application to strongly correlated ground and excited states of molecules自旋限制集合参考的kohn-sham方法: 基本原理及其在分子的强相关基态和激发态中的应用
PREAI
分享收藏
查看更多内容