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

DFT-FE 1.0:一个使用有限元离散化的大规模并行混合CPU-GPU密度泛函理论程序

DFT-FE 1.0: A massively parallel hybrid CPU-GPU density functional theory code using finite-element discretization

Sambit Das, Phani Motamarri, Vishal Subramanian, David M. Rogers, Vikram Gavini

摘要:

我们提出了DFT-FE 1.0,建立在DFT-FE 0.6 [计算。物理通信246,106853(2020)],在众核CPU和混合CPU-GPU计算架构上进行快速准确的大规模密度泛函理论(DFT)计算(达到~ $100,000$电子)。这项工作涉及实空间公式的改进-通过改进静电相互作用的处理,大大提高了计算效率-以及高性能计算方面,包括DFT-FE中所有关键计算内核的GPU加速。我们证明的准确性,通过比较基态能量,离子力和细胞应力对广泛使用的DFT代码获得的基准系统。此外,我们证明了我们的实现,这产生$\sim 20 \times$ CPU-GPU的加速,通过使用GPU加速混合CPU-GPU节点的数值效率。值得注意的是,由于并行缩放的GPU实现,我们获得的壁时间$80-140$秒的完整的基态计算,具有严格的精度,基准系统包含~ $6,000-15,000$电子。

英文摘要:

We present DFT-FE 1.0, building on DFT-FE 0.6 [Comput. Phys. Commun. 246, 106853 (2020)], to conduct fast and accurate large-scale density functional theory (DFT) calculations (reaching ~ $100,000$ electrons) on both many-core CPU and hybrid CPU-GPU computing architectures. This work involves improvements in the real-space formulation -- via an improved treatment of the electrostatic interactions that substantially enhances the computational efficiency -- as well high-performance computing aspects, including the GPU acceleration of all the key compute kernels in DFT-FE. We demonstrate the accuracy by comparing the ground-state energies, ionic forces and cell stresses on a wide-range of benchmark systems against those obtained from widely used DFT codes. Further, we demonstrate the numerical efficiency of our implementation, which yields $\sim 20 \times$ CPU-GPU speed-up by using GPU acceleration on hybrid CPU-GPU nodes. Notably, owing to the parallel-scaling of the GPU implementation, we obtain wall-times of $80-140$ seconds for full ground-state calculations, with stringent accuracy, on benchmark systems containing ~ $6,000-15,000$ electrons.

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

↑