david/flash-attention: flash-attention from https://github.com/Dao-AILab/flash-attention @ changes_for_fp8

mirror da https://github.com/Dao-AILab/flash-attention

Antony Frolov 3566596ad8 Fix typo in RotaryEmbedding forward output type (#666)		1 anno fa
..
__init__.py	ece539abd6 Add __init__.py files to subdirectories for installation	2 anni fa
patch_embed.py	f1a73d0740 Run isort and black on python files	1 anno fa
rotary.py	3566596ad8 Fix typo in RotaryEmbedding forward output type (#666)	1 anno fa