
tilelang
tile-ai
Domain-specific language designed to streamline the development of high-performance GPU/CPU/Accelerators kernels
AI 简介
Tile Language 是一种面向高性能计算的领域专用语言(DSL),专为简化 GPU/CPU/加速器(如 NVIDIA、AMD、Huawei Ascend、Apple Metal)上的底层算子内核开发而设计。它采用类 Python 语法,基于 TVM 编译器框架,支持 GEMM、FlashAttention、稀疏矩阵乘、Dequant 等典型算子的高效表达与自动优化,并集成 CuTe DSL、AscendC、NVRTC、Z3 定理证明等后端与分析能力,兼顾开发效率与硬件级性能。适用于 AI 编译器研发、自定义算子加速、异构硬件适配及高性能库(如推理引擎、训练框架插件)的快速原型构建。
Python
Other6.6k
Stars
629
Forks
39
Watchers
113
Issues
Star 增长
今日0
近 7 天0
近 30 天+57
综合评分70.1
默认分支main