跳到主要内容

Triton

带「Triton」标签的杂记 · 共 1 篇

技术 Triton CUDA

Triton 入门：写第一个 kernel

用 Triton 重写向量加法，理解 program、tl.arange 与 block 启动背后的模型。