AMAZINGINDEX.COM 让生活多一点 Amazing
58.9
VOL. 2026.10
2026.10.02
← 返回 2026.10.02 日报
日报快照 · Daily Snapshot
NO. 005

TileLang:用Python写高性能GPU核函数

#REPO GitHub Trending 2026.10.02
推荐指数 64.0 NO. 005 · 2026.10.02
Stars8,079

TileLang 是基于 TVM 的 Python DSL,让开发者用接近 Python 的语法编写 GEMM、FlashAttention 等高性能 GPU/CPU/NPU 内核,无需手写 CUDA。对需要定制算子但不想深入底层优化的 AI 团队,这是 Triton 之外的新选择。

Domain-specific language designed to streamline the development of high-performance GPU/CPU/Accelerators kernels

TileLang:用Python写高性能GPU核函数

算子优化赛道长期被 Triton 垄断,但 Triton 的学习曲线和调试体验一直是痛点。TileLang 选择站在 TVM 肩膀上而非从头造轮子,这意味着它能直接继承 TVM 的多后端能力(ARM、AMD、NPU),这是 Triton 目前覆盖不足的。

已有团队用它把自定义 Dequant GEMM 做到接近 cuBLAS 性能,而代码量只有 CUDA 的 1/5。如果你在训推一体场景里总被"Triton 不支持这个后端"卡住,或者受够了 Triton 的编译错误信息,TileLang 值得作为备选方案跑一轮 benchmark。

短期风险是生态成熟度:Triton 有 PyTorch 2.0 原生集成和大量社区算子库,TileLang 目前主要靠官方 examples,社区贡献者规模还很小。

Star History
Beta

TVM 生态中的 Pythonic 算子开发 DSL,定位 Triton 替代方案

独特价值:基于成熟 TVM 后端,降低自定义高性能算子开发门槛

竞品:
openai/triton ★ 10.0k Triton 是主流 GPU 算子编译器,TileLang 基于 TVM 提供更 Pythonic 的 DSL
tvm-ai/tvm ★ 12.0k TVM 是底层编译器框架,TileLang 是其上层的领域特定语言封装
查看原文 →