FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awareness
Source
Tri Dao et al.
Author
Tri Dao et al.
Published
Terms in this piece · Glossary
attention — The mechanism that lets a model weigh which earlier words matter for the word it's currently processing — the core operation of a transformer.
Why it matters
The insight that attentionThe mechanism that lets a model weigh which earlier words matter for the word it's currently processing — the core operation of a transformer.Full definition → was memory-bound rather than compute-bound, and that tiling it to avoid round-trips to slow GPU memory makes it several times faster while staying exact. It is in every serving stack you touch, and the clearest worked example of why hardware-aware kernels matter.