Morph MORPH
SIGN IN SIGN UP
tejas / flash-attention

Fast and memory-efficient exact attention

0 0 208 Python

[Cute,Bwd,Sm100] Don't need q_dk_consumer_state

T
Tri Dao committed
5c685eaa7d2bca7eeaae5068f061fabd00fb4d7d
Parent: 7b17cd8