[Bugfix] Fix the import issue in page.py (#102)

Thanks @soodoshll for reporting this issue.
flashinfer-ai · Feb 1, 2024 · 922f0c6 · 922f0c6
1 parent 00aa360
commit 922f0c6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/flashinfer/page.py b/python/flashinfer/page.py
@@ -27,6 +27,8 @@
     else:
         raise e
 
+from .utils import check_kv_layout, TensorLayout
+
 
 def append_paged_kv_cache(
     append_key: torch.Tensor,