Triton 算子

会用 Triton 写 GPU kernel，能用 Python 写出接近手写 CUDA 的性能。

学习流程

• 编程模型和块
• 写向量加法
• 写 softmax
• 写矩阵乘
• 自动调参
• 写一个融合算子

去哪学

• Triton 教程 文档 · 免费