下拉刷新
项目详情
由navbar_avatar分享
repo_avatar
HelloGitHub 评分
0 人评分
用 Python 写高性能 GPU 算子
认领
收藏
分享
8.8k
星数
否
中文
Python
主语言
是
活跃
242
贡献者
395
Issues
是
组织
无
最新版本
930
Forks
无
协议
更多
tilelang image
这是一个专为编写高性能 GPU/CPU/NPU 算子设计的领域专用语言,底层基于 TVM 编译器构建,让开发者不必手写底层代码也能实现 GEMM、FlashAttention 等算子的极致优化。它支持 NVIDIA CUDA、AMD ROCm、Apple Metal 和华为昇腾等多种硬件,内置自动调优和调试可视化工具,并提供 DeepSeek MLA 等大模型算子示例。

评论

评分:
暂无精选评论