tests/test_flash_attn.py · 9d3116addf00e585eaf3f249a6543f5970365288 · gaoqiong / flash-attention · GitLab

Find file Blame History Permalink

Don't enforce bitwise consistency for dq in race condition test · 9d3116ad
Tri Dao authored Nov 13, 2022
```
Since we could be parallelizing over seqlen_k
```
9d3116ad

test_flash_attn.py 54 KB