TileLang 是一种面向 AI 工作负载的领域专用语言(DSL),旨在简化并优化高性能计算内核(如 GEMM、FlashAttention)的开发。它基于 TVM 构建,采用 Pythonic 语法,通过将调度优化与数据流解耦,使开发者能专注于算法逻辑。TileLang 已在包括国产芯片在内的多种 GPU 和加速器上展现出优异的性能。
值得注意的是,TileLang 官方目前并未宣称对 RISC-V 架构的正式支持。但得益于其与 TVM 生态的深度关联,开发者理论上可以借助 TVM 的跨平台编译能力,将其后端扩展至 RISC-V 平台。近日,我们已成功在两款国产 RISC-V 硬件平台——Milk-V Megrez 与进迭时空 K1 上,完成了 TileLang 的源码编译与基础功能验证。