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