flash-attention/tests
Tri Dao e78d509c64 [WIP] Support all head dimensions up to 128 in the Triton bwd
WIP because there seems to be some race conditions for head dimensions other
than 16, 32, 64, 128.
2022-10-31 00:46:22 -07:00
..
test_flash_attn.py [WIP] Support all head dimensions up to 128 in the Triton bwd 2022-10-31 00:46:22 -07:00