标签

Attention

共 1 篇文章

2026

编程

Triton Attention Kernel

Triton 实现 Softmax Attention 算子的基本思路。重点思考怎样不用存完整 attention matrix,仍然精确算出 softmax attention.