PME算法在神威太湖之光上的移植和优化

• 全文链接
• 请求原文

作   者：

林增;  武铮;  安虹;  陈俊仕; 

作者机构：

中国科学技术大学计算机科学与技术学院; 

关键词：

神威太湖之光;  异构众核处理器;  PME;  GROMACS; 

期刊名称：

小型微型计算机系统

i s s n：

1000-1220

年卷期：

2021 年 42 卷 001 期

页   码：

9-14

摘   要：

分子动力学模拟(MD)是一套通过计算机模拟生物体系内分子、原子运动的多体模拟方法.GROMACS是著名的MD应用,能够快速模拟生物及非生物体系运动过程,广泛应用于各高性能平台.作为世界排名第3的超级计算机,神威太湖之光拥有40960块SW26010异构众核处理器,峰值性能达到125.4PFlops.目前太湖之光上已有对GROMACS短程力优化的相关研究,但对于PME(Particle Mesh Ewald)算法未有探索性工作.本文基于申威平台对PME算法展开研究,针对随机访存模式、网格点写写冲突等挑战,提出了基于局部网格序的分块策略、数据重组策略、非线性函数近似等方法进行优化.最终优化后的结果相较于初始版本性能提升了8.85倍,相较于Intel CPU版本提升了1.2倍.本文采用的优化技术也可以为神威太湖之光上其他分子动力学模拟软件和涉及散乱数据插值程序的优化提供借鉴.