david/flash-attention

miroir de https://github.com/Dao-AILab/flash-attention

Auteur	SHA1 Message	Date
Tri Dao	7f5d73a162 Add env var to disable specific hdim	il y a 1 mois
Tri Dao	3e5d77a102 Group instantiations for different hdims together	il y a 1 mois
Tri Dao	fc2fd95a18 Renable FP8 kernels	il y a 2 mois
Tri Dao	64d92bce53 Split PagedKV into separate .cu files to speed up compilation	il y a 2 mois
Tri Dao	586ba914bb Move fwd tile size to a separate file	il y a 2 mois
Tri Dao	018b9af683 Move .cu files to instantiations, use generate_kernels.py	il y a 2 mois