在曙光超级计算机上对第一性原理计算软件LS3DF进行1000万个硅原子模拟

6月前浏览3984

本文摘要：（由ai生成）

JCST期刊发表了一篇关于国产超级计算机上实现大规模第一性原理计算的研究论文。该研究由中国科学院多个研究所和华为技术有限公司合作完成，提出了一种线性标度三维分块算法（LS3DF），成功处理了高达1000万个硅原子的系统，实现了高峰值性能。该算法通过算法和系统层面的优化，显著提高了计算效率和可扩展性，为未来更大规模超级计算机上的电子结构模拟提供了技术基础。论文作者来自高性能计算、大规模并行计算、第一性原理计算等多个领域，展示了跨学科合作的成果。

JCST于2024年第1期出版一篇关于在国产超算上实现大规模第一性原理计算的研究论文。该论文提出了一种可大规模扩展且高效的第一性原理计算软件LS3DF的实现，使其能够在曙光超级计算机上处理高达1000万个硅原子的系统，并达到34.8 PFLOPS的峰值性能。本篇论文由来自中国科学院计算技术研究所、中国科学院大学、中国科学院半导体研究所、盐城工学院和华为技术有限公司多位作者共同撰写。

研究动机

随着摩尔定律的推进，半导体器件的尺寸缩小至纳米级别。在这种尺度下，量子效应对半导体器件仿真中的影响变得至关重要，因此带来了大规模电子结构计算的挑战。当前电子结构计算主要依赖密度泛函理论，但该方法在处理大规模系统时面临着高昂的计算成本。

为应对这一问题，本研究在国产曙光超级计算机上实现了一种大规模、高效的线性标度三维分块算法（LS3DF）。该文通过对算法和系统层面的优化，成功在一千万原子体系的模拟中表现出极佳的可扩展性和良好的计算效率，为仿真模拟下一代半导体器件提供关键支持。

研究方法

本研究通过一系列算法级和系统级优化方法，成功实现了LS3DF方法的高效率和大规模应用。

在算法层面，对原有的共轭梯度迭代算法进行了优化，改进了迭代方向，从而加快了算法的收敛速度。此外，还采纳了混合精度的计算策略以缩短计算时间，同时通过理论分析保证了算法准确性的不受影响。

在系统层面，引入粗粒度并行处理方法，针对原先算法的细粒度划分导致大量通信代价的问题，如图1所示，通过改变算法的划分粒度以减少数据传输所需的时间。如图2所展示的，本文通过实施多流三维快速傅里叶变换方法，可更充分利用曙光深度计算单元的计算资源。此外，本文还采取了其它系统优化措施，例如核融合和冗余计算移除，进一步提高异构计算机的计算效率。

通过以上两个层面的优化策略，LS3DF方法能够有效处理高达1000万个硅原子的复杂系统，并在峰值性能上取得显著提升。这项工作对于未来在更大规模超级计算机上实现整个半导体器件的电子结构模拟工作具有深远的意义。

图1. LS3DF方法的细粒度并行与粗粒度并行

图2. 多流三维快速傅里叶变换方法

实验和结果

本研究首先对LS3DF算法在小规模体系上的效率进行了测试，并在图3中展示了结果。经过优化的LS3DF算法在效率上比原始异构版本累计提高了3.1倍。

同时，本文在大规模体系上进行了算法的扩展性测试。图4展示强扩展性结果：当计算一百万原子体系的节点从400个增加到3200个时，并行效率仍能维持在81%。图5则展示了弱扩展性结果：当计算原子体系规模从8000增至1000万（计算节点相应从3增加至3800）时，算法的扩展效率达到了98.72%，并实现了峰值性能的21.2%（34.8 PFLOPS）。