18.7.1 FlashAttention V1 详解#

FlashAttention V1 详解深入剖析 FlashAttention V1 的核心原理与实现细节,理解如何通过 Tiling + Online Softmax 将 Attention 的额外显存占用从 $O(N^2)$ 降至 $O(N)$,同时大幅减少 HBM 访问量,实现无精度损失的加速。

登录后才可看