上海
品玩2月11日訊，摩爾線程正式開源TileLang-MUSA項目，實現(xiàn)對TileLang編程語言的完整支持。該項目已在摩爾線程多代全功能GPU上完成驗證，旨在通過高層抽象與編譯器優(yōu)化，顯著降低國產(chǎn)算力平臺的AI與高性能計算開發(fā)門檻。
TileLang是一款基于張量分塊抽象的領域特定語言，采用類Python前端，允許開發(fā)者以接近數(shù)學公式的形式描述計算意圖。編譯器可自動完成底層硬件優(yōu)化，在保持高性能的同時降低編程復雜度。TileLang-MUSA項目連接了TileLang語法與摩爾線程MUSA硬件架構，能夠自動調(diào)用Tensor Core、實現(xiàn)多級數(shù)據(jù)流水線，并支持Warp級并行優(yōu)化。
實際測試顯示，在摩爾線程MTT S5000上，使用TileLang-MUSA開發(fā)關鍵算子（如GEMM和FlashAttention-3）相比手寫代碼可減少約90%代碼量，同時性能可達手寫優(yōu)化版本的85%至95%。此舉為開發(fā)者提供了高效工具鏈，助力構建開放、易用的國產(chǎn)算力應用生態(tài)。
特別聲明：以上內(nèi)容(如有圖片或視頻亦包括在內(nèi))為自媒體平臺“網(wǎng)易號”用戶上傳并發(fā)布，本平臺僅提供信息存儲服務。
Notice: The content above (including the pictures and videos if any) is uploaded and posted by a user of NetEase Hao, which is a social media platform and only provides information storage services.
