david/flash-attention: flash-attention from https://github.com/Dao-AILab/flash-attention @ fa3-kvcache-gqa

peilaus alkaen https://github.com/Dao-AILab/flash-attention

Tri Dao 0e8c46ae08 Run isort and black on test files		1 vuosi sitten
..
test_rotary.py	0e8c46ae08 Run isort and black on test files	1 vuosi sitten