bugfix: fix append_paged_kv_cache test (#625)

Co-authored-by: junq <junq@nvidia.com>
flashinfer-ai · Nov 21, 2024 · 9cba9fb · 9cba9fb
1 parent ba1d8c3
commit 9cba9fb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_page.py b/tests/test_page.py
@@ -41,7 +41,7 @@ def test_append_paged_kv_cache(contiguous):
     kv_last_page_len = torch.tensor([13, 8, 9, 6], dtype=torch.int32, device="cuda:0")
     batch_indices, positions = flashinfer.get_batch_indices_positions(
         kv_append_indptr,
-        flashinfer.get_seq_lens(kv_append_indptr, kv_last_page_len, page_size),
+        flashinfer.get_seq_lens(kv_page_indptr, kv_last_page_len, page_size),
         nnz_kv,
     )