examples/linear_attention/example_linear_attn_fwd.py · b7dfdb39b91532df74cc5032d4460df867cbc0cb · 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_fwd.py 4.66 KB