Morph MORPH
SIGN IN SIGN UP
tejas / flash-attention

Fast and memory-efficient exact attention

0 0 208 Python

[Cute,Bwd,Sm100] Combine pipeline_dK and pipeline_dV into one

T
Tri Dao committed
12e1c0498cf520458c290064e5493dc92f02a697
Parent: 2187695