From 059cfced143b3c7b88e4f4cf589f7a188c1f7037 Mon Sep 17 00:00:00 2001 From: Zihao Ye Date: Sun, 24 Nov 2024 03:11:03 +0000 Subject: [PATCH] upd --- tests/test_shared_prefix_kernels.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/test_shared_prefix_kernels.py b/tests/test_shared_prefix_kernels.py index 9ef2103c..5a8bbf2c 100644 --- a/tests/test_shared_prefix_kernels.py +++ b/tests/test_shared_prefix_kernels.py @@ -44,6 +44,7 @@ def warmup_jit(): [False], # use_logits_soft_caps [False], # allow_fp16_qk_reductions ) + + [flashinfer.page.get_page_module, []] ) except Exception as e: # abort the test session if warmup fails