examples/deepseek_nsa/example_tilelang_nsa_fwd_varlen.py · 18889821dd3ca90f215122e55c99985b55a0cd5c · OpenDAS / tilelang

"mmdet/models/task_modules/tracking/interpolation.py" did not exist on "1189a8adcc55e3a2530ce8b716c55a67b2d53af1"

[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_tilelang_nsa_fwd_varlen.py 14.3 KB

Replace example_tilelang_nsa_fwd_varlen.py