minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

[Test] Fix Q8KV8 sparse-prefill pool fixture after page-size rename (#39019)

Browse Source
This commit is contained in:
Jialin Ouyang
2026-09-10 22:22:35 -07:00
committed by GitHub
parent 2018c64e22
commit 6182d45524
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/registered/kernels/ops/attention/test_q8kv8_sparse_prefill_backend.py
+2 -2
View File
@@ -76,7 +76,7 @@ class _TokenToKVPool:
extra_key_buffer if extra_key_buffer is not None else swa_key_buffer 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.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: def get_swa_key_buffer_radix(self, layer_id: int) -> torch.Tensor:
_ = layer_id _ = layer_id
@@ -84,7 +84,7 @@ class _TokenToKVPool:
def get_extra_key_page_size(self, layer_id: int) -> int: def get_extra_key_page_size(self, layer_id: int) -> int:
_ = layer_id _ = layer_id
return self.swa_window_size return self.swa_page_size
def get_extra_key_buffer(self, layer_id: int) -> torch.Tensor: def get_extra_key_buffer(self, layer_id: int) -> torch.Tensor:
_ = layer_id _ = layer_id