david/flash-attention

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

Author	SHA1 Message	Date
Tri Dao	76f14c61c9 Tune fwd tile sizes for Sm86 and Sm89	2 months ago
Tri Dao	51484a7b56 Make backward epilogue work for Sm80	2 months ago
Tri Dao	659a631f4c Rename bwd classes to include Sm90 suffix	2 months ago
Tri Dao	a901c7eeda Make Sm80 forward pass work with persistent scheduler	2 months ago
Tri Dao	5171269dab Implement forward pass for Sm80	2 months ago