请你实现一个FlashAttention算法的核心代码，并解释其工作原理。

Question

Accepted Answer

考察点：考察对FlashAttention算法的理解深度与编码实现能力。回答思路：能正确实现分块计算与在线softmax 说明IO感知的分块策略 解释与标准注意力相比的优势 处理数值稳定性与重计算

考察说明