examples/deepseek_nsa/example_triton_nsa_bwd.py · e84b24bc6a6b4bbfece53dcc221791bd169e09cf · OpenDAS / tilelang

"alphafold/vscode:/vscode.git/clone" did not exist on "db039ef4230081808aff4d3fcf8eb83c52348004"

[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