squall/flash-attention
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
008951f1d9
flash-attention/tests
History
Tri Dao 008951f1d9 Support all head dimensions up to 128 in the Triton fwd
2022-10-30 22:10:48 -07:00
..
test_flash_attn.py Support all head dimensions up to 128 in the Triton fwd 2022-10-30 22:10:48 -07:00