跳转到内容

Triton:块级编程

Triton 把编程粒度从线程提升到块,由编译器处理共享内存分配与同步。本章讲 Triton 的编程模型、自动调优、以及它在融合算子与自定义注意力上的典型用法与局限。

  • 块级抽象大幅降低编写融合内核的门槛
  • 自动调优搜索空间需要人工限定
  • Triton 内核在不同 GPU 代际上的可移植性