examples/deepseek_nsa/example_triton_nsa_bwd.py · 18ab72c90bf752536baa9eba520dbae8f8abe1e0 · OpenDAS / tilelang

[BugFix] Fix import error in nsa examples when `fla.__version__ >=0.2.1` (#579) · 18ab72c9

Tong WU authored Jun 16, 2025

* Update FLA import path for `prepare_token_indices`

* Update FLA import path for `prepare_token_indices`

* Compare versions via packaging.version.parse

18ab72c9

example_triton_nsa_bwd.py 35.5 KB

Replace example_triton_nsa_bwd.py