品玩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%。此举为开发者提供了高效工具链,助力构建开放、易用的国产算力应用生态。
打开网易新闻 查看精彩图片
热门跟贴