第 33 章 TileKernels(上):一份算法,两个后端
/code/33-tilekernels-two-backends/01-topk-gate-minimal
本章代码
正文里引用到的示例都在这里,可直接在浏览器中阅读;代码与教材仓库保持一致。
- README.md2541 字节
- topk_gate_asc_min.py7954 字节
- topk_gate_cuda_min.py4057 字节
tilelang-tutorial.huandu.me
/code/33-tilekernels-two-backends/01-topk-gate-minimal
正文里引用到的示例都在这里,可直接在浏览器中阅读;代码与教材仓库保持一致。