squall/flash-attention
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
08e9847176
flash-attention/tests
History
dan_the_3rd 011ec323d6
Support MQA + MP for decoding (#490) 
Co-authored-by: danthe3rd <danthe3rd>
2023-08-30 10:29:54 -07:00
..
layers Run isort and black on test files 2023-08-18 20:59:35 -07:00
losses Run isort and black on test files 2023-08-18 20:59:35 -07:00
models Support MQA + MP for decoding (#490) 2023-08-30 10:29:54 -07:00
modules Run isort and black on test files 2023-08-18 20:59:35 -07:00
ops Run isort and black on test files 2023-08-18 20:59:35 -07:00
pyproject.toml Move pyproject.toml to flash-attn and tests dir to avoid PEP 517 2023-08-25 15:05:28 -07:00
test_flash_attn.py Implement splitKV attention 2023-08-29 00:58:29 -07:00
test_rotary.py Run isort and black on test files 2023-08-18 20:59:35 -07:00