david/flash-attention

огледало от https://github.com/Dao-AILab/flash-attention

Автор	SHA1 Съобщение	Дата
Ying Zhang	496fdc4f6c Add seqused_q in fwd / bwd and seqused_k in bwd.	преди 5 месеца
Jay Shah	32792d37ec add missing if condition for key_padding_mask in test_util.py	преди 5 месеца
Ying Zhang	53537da422 add a unittest	преди 5 месеца
Ying Zhang	dfe1a59e4b Add var-seq-len to FA3 fp16 / bf16 fwd (#1072)	преди 6 месеца