Featured image of post FlashAttention: Move Less, Not Compute Less

FlashAttention: Move Less, Not Compute Less

FlashAttention neither approximates attention nor removes its quadratic arithmetic. It uses tiling, online softmax, and recomputation to avoid materializing the attention matrix. This article explains the GPU memory hierarchy behind the speedup and the evolution through FA2 and FA3.

(1 - 6)
Enter Press Enter to jump