flash-attention/tests
2022-10-24 16:04:21 -07:00
..
test_flash_attn.py Support all head dims that are multiples of 8, up to 128 2022-10-24 16:04:21 -07:00