Settings

Theme

Jagged Flash Attention Optimization

shaped.ai

24 points by tullie · 4 comments

Reader
2 threads
platers

Flash attention natively supports packing multiple variable length sequences into a single call, what is the advantage of jagged flash attention?

  • bbstats

    If only there was a link to a page somewhere that could answer this question for you.

CapsAdmin

See also

https://github.com/thu-ml/SageAttention and https://github.com/thu-ml/SpargeAttn

Keyboard Shortcuts

j
Next item
k
Previous item
o / Enter
Open selected item
?
Show this help
Esc
Close modal / clear selection