品玩2月11日訊,摩爾線程正式開源TileLang-MUSA項目,實現對TileLang編程語言的完整支持。該項目已在摩爾線程多代全功能GPU上完成驗證,旨在通過高層抽象與編譯器優化,顯著降低國產算力平臺的AI與高性能計算開發門檻。
TileLang是一款基于張量分塊抽象的領域特定語言,采用類Python前端,允許開發者以接近數學公式的形式描述計算意圖。編譯器可自動完成底層硬件優化,在保持高性能的同時降低編程復雜度。TileLang-MUSA項目連接了TileLang語法與摩爾線程MUSA硬件架構,能夠自動調用Tensor Core、實現多級數據流水線,并支持Warp級并行優化。
實際測試顯示,在摩爾線程MTT S5000上,使用TileLang-MUSA開發關鍵算子(如GEMM和FlashAttention-3)相比手寫代碼可減少約90%代碼量,同時性能可達手寫優化版本的85%至95%。此舉為開發者提供了高效工具鏈,助力構建開放、易用的國產算力應用生態。
![]()
特別聲明:以上內容(如有圖片或視頻亦包括在內)為自媒體平臺“網易號”用戶上傳并發布,本平臺僅提供信息存儲服務。
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.