tile-ai/tilelang：用 Python 语法写高性能芯片内核
来源：GitHub — 2026-10-02 · ⭐ 8,032 · Python
📋 概述
TileLang 是一门简洁的领域专用语言，目标是让 GEMM、反量化 GEMM、FlashAttention、LinearAttention 这类高性能内核写起来不那么痛苦：上层是贴近 Python 的语法，底下是架在 TVM 之上的编译器基础设施，开发者既不用放弃底层优化，也不必先成为 CUDA 专家。它今日新增百余星，恰逢一条重要更新——官方宣布原生支持华为昇腾 950 NPU，带原生代码生成、自动调度与同步、SIMD/SIMT 向量编程，并给出 GEMM 与 FlashAttention 的昇腾示例。此前还开源了 TileLang LSP，为缓冲区形状、数据类型、存储域与推断布局提供内联提示。Python 实现，配官方文档站与 Discord 社区。
🔑 核心要点
- 定位是领域专用语言，专攻 GEMM、FlashAttention 这类高性能内核的编写
- 上层用贴近 Python 的语法表达，底层编译器架在 TVM 之上，兼顾生产率与低层优化
- 2026-09-30 起原生支持华为昇腾 950 NPU，含原生代码生成、自动调度与同步
- 昇腾后端同时覆盖 SIMD/SIMT 向量编程，并附 GEMM、FlashAttention 示例
- 已开源 TileLang LSP，为缓冲区形状、dtype、作用域与推断布局提供内联提示
- Python 实现、官方文档站 tilelang.com，另有配套 puzzle 仓库与 Discord 社区
💡 金句
把内核优化从「先成为 CUDA 专家」变成「先把算法写清楚」。
👍 0
👎 0
← 返回 GitHub Trending 首页