据摩尔线程官微2月10日消息,近日,摩尔线程-U(688795.SH)正式开源TileLang-MUSA项目,实现对TileLang编程语言的完整支持。该项目已成功在摩尔线程多代全功能GPU上完成功能验证与特性开发,旨在通过高层抽象与编译器优化,大幅降低开发门槛,为国产算力平台提供更高效的AI与高性能计算开发体验。
据介绍,TileLang是一款基于张量分块(Tiling)抽象的高性能AI算子编程语言,属于领域特定语言(DSL)。它采用声明式语法与类Python前端,使开发者能够以接近数学公式的形式描述计算意图,并由编译器自动完成循环优化、内存调度与代码生成,在保持底层性能的同时大幅降低GPU及异构计算平台的编程复杂度。
在实际应用中,TileLang通过三大核心作用显著提升GPU计算的开发效率:通过高级抽象降低开发门槛,开发者无需深入底层硬件知识即可生成高性能代码;具备跨平台能力,实现“一次编写、多架构运行”,有效解决多元算力生态的适配难题;编译器自动执行Layout推导、线程映射、Warp特化、流水线排布、内存优化等复杂优化,在保障性能的同时提升开发效率。
TileLang已广泛应用于多个关键领域,例如:在AI与机器学习中,研究人员可用其快速定义新型算子(如注意力机制);在科学计算中,可便捷移植大型数值模拟程序至不同硬件平台;对硬件厂商而言,TileLang可作为构建芯片软件生态的基础工具链。





京公网安备 11011402013531号