david/flash-attention

Mirror von https://github.com/Dao-AILab/flash-attention

Autor	SHA1 Nachricht	Datum
Tri Dao	1e3208566a Tune tile sizes for compilation	vor 1 Monat
Tri Dao	df5fe55264 Change tile sizes for Sm8x to reduce stack frame	vor 1 Monat
Tri Dao	5f525322ec Only pass sm_90a compile flag to Sm90 kernels, same w Sm89 kernels	vor 1 Monat
Tri Dao	180ff782dd Template for Sm86	vor 1 Monat
Tri Dao	7a802796e1 Big refactor and update	vor 1 Monat