examples/linear_attention/example_retention_fwd.py · 4d54854be46c6ace0f53055897517fd9fd2359ef · 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_retention_fwd.py 4.55 KB