摩爾線程官方在2月10日宣布,近期正式開源了TileLang-MUSA項目,為TileLang編程語言提供全面支持。該項目已在摩爾線程多代全功能GPU上完成功能驗證與特性開發,旨在通過高層抽象與編譯器優化,大幅降低開發門檻,為國產算力平台提供更高效的AI與高性能計算開發體驗。
據了解,TileLang是一款基於張量分塊抽象的高性能AI算子編程語言,屬於領域特定語言(DSL)範疇。它採用聲明式語法與類Python前端,使開發者能夠以貼近數學公式的形式描述計算意圖,並由編譯器自動完成循環優化、內存調度與代碼生成,在保持底層性能的同時,顯著降低GPU及異構計算平台的編程複雜度。
在實際應用中,TileLang通過三大核心作用顯著提升GPU計算的開發效率:藉由高級抽象降低開發門檻,開發者無需深入底層硬件知識即可生成高性能代碼;具備跨平台能力,實現“一次編寫、多架構運行”,有效解決多元算力生態的適配難題;編譯器自動執行Layout推導、線程映射、Warp特化、流水線排布、內存優化等複雜優化,在保障性能的同時提升開發效率。
TileLang已廣泛應用於多個關鍵領域,例如:在AI與機器學習中,研究人員可用其快速定義新型算子(如注意力機制);在科學計算中,可便捷移植大型數值模擬程序至不同硬件平台;對硬件廠商而言,TileLang可作為構建芯片軟件生態的基礎工具鏈。
