squall/flash-attention
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
b910bf14c1
flash-attention/tests
History
Tri Dao b910bf14c1 Support arbitrary seqlens (both q & k) in Triton bwd
2022-10-30 21:50:53 -07:00
..
test_flash_attn.py Support arbitrary seqlens (both q & k) in Triton bwd 2022-10-30 21:50:53 -07:00