examples/flash_attention/example_gqa_fwd_bshd.py · c5bbc608555014e7b0b7108bc0a5eebb1522768f · OpenDAS / tilelang · GitLab

Find file Blame History Permalink

[Examples] Expand tuning configurations for FlashAttention example (#204) · 8678aac0
Chenghua authored Mar 14, 2025
```
* [Example] Modify tuning configurations for FlashAttention example

* [Examples] formatting example_gqa_fwd_bshd.py
```
8678aac0

example_gqa_fwd_bshd.py 11.3 KB