squall/flash-attention
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
35d589fa81
flash-attention/tests
History
Tri Dao 52fb4b729b Fix #54: set device for multi-GPU case
2022-10-16 12:51:26 -07:00
..
test_flash_attn.py Fix #54: set device for multi-GPU case 2022-10-16 12:51:26 -07:00