下拉刷新
Repository Details
Shared bynavbar_avatar
repo_avatar
HelloGitHub Rating
0 ratings
Writing High-Performance GPU Operators in Python
Claim
Collect
Share
8.8k
Stars
No
Chinese
Python
Language
Yes
Active
242
Contributors
395
Issues
Yes
Organization
None
Latest
930
Forks
None
License
More
tilelang image
This is a domain-specific language designed for writing high-performance GPU/CPU/NPU operators. Built on top of the TVM compiler, it lets developers achieve highly optimized implementations of operators such as GEMM and FlashAttention without hand-writing low-level code. It supports multiple hardware backends, including NVIDIA CUDA, AMD ROCm, Apple Metal, and Huawei Ascend, and comes with built-in auto-tuning and debugging visualization tools, plus example operators for large models such as DeepSeek MLA.

Comments

Rating:
No comments yet