IT之家11月1日报道,在北京大学鲲鹏昇腾科教创新卓越中心支持下,北京大学力学与工程科学学院陈默涵研究员团队完成ABACUS在鲲鹏920全系列平台上的适配,并把19类材料领域核心算子抽出来逐项优化。
团队为鲲鹏920的NEON、SVE、SVE512及矩阵运算单元分别适配编译路径,新增USE_KML选项接入鲲鹏数学库;NEON、SVE和矩阵运算三个版本均通过平面波基组、LCAO、sDFT、DeePKS等集成测试,数值结果与业界主流实现一致。
在28核、28进程×1线程配置下,SVE版本在标准测试集上快1.3至3.9倍;单节点满核启用矩阵运算单元最高快10.7倍。随机轨道DFT算例中,SVE版本快2.8至3.1倍,矩阵运算指令集版本快6.1至20.1倍。相关代码已推送到GitHub开源社区。
