WIP because there seems to be some race conditions for head dimensions other than 16, 32, 64, 128. |
||
|---|---|---|
| .. | ||
| test_flash_attn.py | ||
WIP because there seems to be some race conditions for head dimensions other than 16, 32, 64, 128. |
||
|---|---|---|
| .. | ||
| test_flash_attn.py | ||