fix pytest

yzh119 · yzh119 · commit c27d1fa0442d · 2025-08-10T05:36:06.000-04:00
diff --git a/flashinfer/prefill.py b/flashinfer/prefill.py
@@ -36,7 +36,6 @@
     setup_metainfo_loader,
     trtllm_gen_fmha_module,
 )
-from .cudnn import cudnn_batch_prefill_with_kv_cache
 from .page import block_sparse_indices_to_vector_sparse_offsets, get_seq_lens
 from .quantization import packbits, segment_packbits
 from .utils import (

Original file line number	Diff line number	Diff line change
`@@ -36,7 +36,6 @@`
`36`	`36`	`setup_metainfo_loader,`
`37`	`37`	`trtllm_gen_fmha_module,`
`38`	`38`	`)`
`39`		`-from .cudnn import cudnn_batch_prefill_with_kv_cache`
`40`	`39`	`from .page import block_sparse_indices_to_vector_sparse_offsets, get_seq_lens`
`41`	`40`	`from .quantization import packbits, segment_packbits`
`42`	`41`	`from .utils import (`