Skip to content

Commit

Permalink
misc: add page ops to shared-prefix kernel unittest warmup function (#…
Browse files Browse the repository at this point in the history
…630)

In #629 , page ops are missing in shared-prefix kernel unittest's warmup
function
  • Loading branch information
yzh119 authored Nov 24, 2024
1 parent 8f5f349 commit 6ec4db3
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions tests/test_shared_prefix_kernels.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 6ec4db3

Please sign in to comment.