examples/linear_attention/example_linear_attn_bwd.py · 0fd3a3e860a56490a52b330ac49de4f7a8615e78 · OpenDAS / tilelang · GitLab

Find file Blame History Permalink

[Dev] Update linear attention examples to enhance performance on Hopper GPUs (#621) · 0fd3a3e8
Tong WU authored Jul 09, 2025
```
* Tune linear attention examples on H100

* Add retnet fwd kernel

* fix lint
```
0fd3a3e8

example_linear_attn_bwd.py 7.58 KB