Fix tensor mismatch error in sepc + topk > 1 + page_size > 1 (#14874)

2025-12-15 01:10:58 +08:00
parent 47633c191f
commit 997ea57eaf
2 changed files with 4 additions and 2 deletions
@@ -228,6 +228,7 @@ class TestEAGLERadixCache(CustomTestCase):
            # Chunked prefill & Page Size > 1
            {**self.BASE_CONFIG, "chunked_prefill_size": 64, "page_size": 4},
            {**self.BASE_CONFIG, "page_size": 4},
+            {**self.BASE_CONFIG, "cuda_graph_bs": [5], "page_size": 4},
            # Preferred by some kernels
            {**self.BASE_CONFIG, "page_size": 64},
            # Disable CUDA Graph