From 6182d4552476bb9d317e4fbf169e0dded557ebfb Mon Sep 17 00:00:00 2001 From: Jialin Ouyang Date: Thu, 10 Sep 2026 22:22:35 -0700 Subject: [PATCH] [Test] Fix Q8KV8 sparse-prefill pool fixture after page-size rename (#39019) --- .../ops/attention/test_q8kv8_sparse_prefill_backend.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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 c74f30e65..739cf72bd 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 @@ -76,7 +76,7 @@ class _TokenToKVPool: extra_key_buffer if extra_key_buffer is not None else swa_key_buffer ) self.full_to_swa_index_mapping = full_to_swa_index_mapping - self.swa_window_size = page_size + self.swa_page_size = page_size def get_swa_key_buffer_radix(self, layer_id: int) -> torch.Tensor: _ = layer_id @@ -84,7 +84,7 @@ class _TokenToKVPool: def get_extra_key_page_size(self, layer_id: int) -> int: _ = layer_id - return self.swa_window_size + return self.swa_page_size def get_extra_key_buffer(self, layer_id: int) -> torch.Tensor: _ = layer_id