FlashAttention 动效 02
小例子:4 个 query、6 个 key(分成 3 块,每块 2 个)。每加载一个 K/V 块,就更新一次行状态 m(运行最大值)、ℓ(指数和)和未归一化输出 Õ。点「下一步」逐块推进。