david/flash-attention

mirrorاز https://github.com/Dao-AILab/flash-attention

نویسنده	SHA1 پیام	تاریخ
Tri Dao	76f14c61c9 Tune fwd tile sizes for Sm86 and Sm89	2 ماه پیش
Tri Dao	51484a7b56 Make backward epilogue work for Sm80	2 ماه پیش
Tri Dao	659a631f4c Rename bwd classes to include Sm90 suffix	2 ماه پیش
Tri Dao	a901c7eeda Make Sm80 forward pass work with persistent scheduler	2 ماه پیش
Tri Dao	5171269dab Implement forward pass for Sm80	2 ماه پیش