examples/flash_attention/example_mha_fwd_bhsd.py · b12a63cfab25f8cca530d1e95f05f189298ec507 · OpenDAS / tilelang · GitLab

Find file Blame History Permalink

[Bugfix] Ensure correct handling for cases where `seq_q<seq_kv` in flash attention examples (#864) · b12a63cf
Tong WU authored Sep 23, 2025
```
* fix flash attention examples  for `seqlen_q<seqlen_kv` cases

* lint
```
b12a63cf

example_mha_fwd_bhsd.py 9.13 KB