【ITBEAR】摩爾線程近日宣布，已正式將高性能線性代數模板庫MUTLASS開源，旨在助力開發者更高效地為摩爾線程GPU的MUSA Core與Tensor Core等單元進行編程。此舉有望加快基于國產GPU的算子開發與算法創新步伐。
MUTLASS的開源，是摩爾線程在開放技術資源方面的又一重要舉措。此前，該公司已陸續開源了OpenCV-MUSA計算機視覺庫、MooER音頻理解大模型以及vLLM-MUSA大語言模型高速推理框架，顯示出其對開源社區的支持與投入。
在數值計算和深度學習領域，矩陣乘法及其變種是構建上層復雜應用的基礎。然而，為實現更高的算子融合效率或更創新的算法，開發者們常需超越標準計算接口的限制。MUTLASS應運而生，專為滿足這一需求而設計。
作為摩爾線程針對自研MUSA架構優化的高性能計算庫，MUTLASS基于開源模板庫CUTLASS進行了適配和定制化開發。它提供了一系列高性能的C++模板組件，并采用分層分解及數據搬運策略，以確保性能的充分發揮。
此次開源的版本中，摩爾線程為CuTe后端庫增加了第三代MUSA架構的MMA計算原語，支持多種數據精度，包括TF32、FP16、BF16和INT8。這些功能為開發者提供了更廣泛的選擇和靈活性。
借助MUTLASS，開發者們可以靈活復用不同層級的模板組件，并根據需要修改實現細節，以較低的成本實現定制化的高性能算子。這不僅有助于在摩爾線程全功能GPU上充分釋放性能，還為算法創新提供了更多可能。
摩爾線程表示，將繼續致力于優化MUTLASS的性能，并不斷引入新功能，以滿足開發者日益增長的需求。
