跳到主要内容

2 篇文档带有标签「triton」

查看所有标签

Triton 编译器源码级实战

沿 Python DSL → AST → TTIR → TTGIR → LLVM IR/PTX 的多级 lowering 链路拆解 Triton 编译器,并亲手 dump 各级 IR 观察 fused softmax kernel 的编译全过程。

极致算子实战案例 (SOTA Kernel)

拆解 FlashAttention 家族(FA-1/2/3)的 tiling + online softmax + 重计算三板斧,量化与 MoE 内核工程,并用 Triton 手写简化 FlashAttention 与 PyTorch SDPA 对数值、对速度。