From c608f9bf7581ea87dd2efccfc34093e22b3e8810 Mon Sep 17 00:00:00 2001
From: "Po-Han Huang (NVIDIA)" <53919306+nvpohanh@users.noreply.github.com>
Date: Thu, 27 Aug 2026 16:00:22 +0800
Subject: [PATCH] [CI] Fix Q8KV8 sparse prefill test fixture (#36639)

---
 .../kernels/ops/attention/test_q8kv8_sparse_prefill_backend.py   | 1 +
 1 file changed, 1 insertion(+)

diff --git a/test/registered/kernels/ops/attention/test_q8kv8_sparse_prefill_backend.py b/test/registered/kernels/ops/attention/test_q8kv8_sparse_prefill_backend.py
index ca2adb70e..7347bada1 100644
--- a/test/registered/kernels/ops/attention/test_q8kv8_sparse_prefill_backend.py
+++ b/test/registered/kernels/ops/attention/test_q8kv8_sparse_prefill_backend.py
@@ -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)