[Bugfix][DSv4] Seed the -1 sentinel in the prefill sparse index workspace (#55299)

Signed-off-by: aoshen02 <aoshen02@users.noreply.github.com> Co-authored-by: aoshen02 <aoshen02@users.noreply.github.com>
2026-09-05 18:31:04 +08:00
parent 2902ca17e3
commit 7fbd44cbe0
1 changed files with 1 additions and 0 deletions
@@ -561,6 +561,7 @@ def combine_topk_swa_indices(
        )
    else:
        combined_indices, combined_lens = out
+        combined_indices.fill_(-1)

    _COMBINE_TOPK_SWA_INDICES_KERNEL(
        combined_indices,