近日,在北京大學 鯤鵬昇騰科教創新卓越中心提供的算力支持下,由北京大學化學與分子工程學院副研究員楊立江主導的"分子模擬程序SPONGE在鯤鵬上的適配與優化"研究課題取得重大突破。該項目依托鯤鵬處理器的高性能計算能力,成功完成了SPONGE分子動力學軟件的并行計算改造,實現了性能超越國際主流GPU水平及并行效率的提升,標志著我國分子動力學模擬軟件創新能力邁上了新臺階。
作為可廣泛應用于生物醫藥和材料科學領域強大計算工具,SPONGE的卓越性能主要體現在兩個方面。首先在并行效率方面,團隊利用鯤鵬處理器的多核架構和高并發性能,設計出高效的區域分解并行算法。通過鯤鵬KML數學庫,優化SPONGE中的關鍵數學運算,提升計算密集型任務的執行效率,使程序在多核環境下的并行效率提升至80%。
同時,在模型性能方面,鯤鵬處理器的矩陣加速計算能力提升了模型的推理速度。通過使用畢昇編譯器生成高性能的機器碼,提升模型算子性能,進而對SPONGE目前已有的深度分子模型進行優化,使SPONGE運行速度得到了進一步的提升。經過優化后的SPONGE在單片鯤鵬處理器上的性能達到了國際主流GPU的102%。
基于以上優勢,SPONGE軟件將陸續服務于高校、研究所等科研單位,推動物理學、化學、生命科學和材料科學的研究。截至目前,優化后的SPONGE軟件已成功應用于北京昌平實驗室主導的生物大分子相互作用模擬研究,為生命科學領域的科研工作提供了強有力的工具支持。
該科研創新成果不僅為多個跨學科研究提供強有力的技術支撐,更展現了自主算力在科學計算領域的強大潛力。目前,相關代碼已通過第三方平臺部分開放,將進一步推動行業應用。未來,北京大學 鯤鵬昇騰科教創新卓越中心將持續深化產學研合作,推動更多前沿科技在國內計算平臺上的創新應用,為中國計算產業的自主創新注入強勁動力。