[CI] Fix Q8KV8 sparse prefill test fixture (#36639)

2026-08-27 01:00:22 -07:00
parent b8a6adadfe
commit c608f9bf75
1 changed files with 1 additions and 0 deletions
@@ -203,6 +203,7 @@ def _make_sparse_prefill_case(
        ),
        page_size=page_size,
    )
+    backend.token_to_kv_pool = token_to_kv_pool

    generator = torch.Generator(device=device)
    generator.manual_seed(11)