david/flash-attention

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

Auteur	SHA1 Message	Date
Ying Zhang	8cbc8a042f small fixes	il y a 3 mois
Ying Zhang	cdbbe844b1 minor changes to unpad_input test util func	il y a 3 mois
Ying Zhang	db80387343 Add seqused_q in fwd / bwd and seqused_k in bwd.	il y a 3 mois
Jay Shah	32792d37ec add missing if condition for key_padding_mask in test_util.py	il y a 3 mois
Ying Zhang	53537da422 add a unittest	il y a 4 mois
Ying Zhang	dfe1a59e4b Add var-seq-len to FA3 fp16 / bf16 fwd (#1072)	il y a 4 mois