From 922f0c6ec5142e6e31171ed5fda594bde0f5e3e5 Mon Sep 17 00:00:00 2001
From: Zihao Ye <expye@outlook.com>
Date: Thu, 1 Feb 2024 11:58:41 +0800
Subject: [PATCH] [Bugfix] Fix the import issue in `page.py` (#102)

Thanks @soodoshll for reporting this issue.
---
 python/flashinfer/page.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/python/flashinfer/page.py b/python/flashinfer/page.py
index d7944a782..876e98229 100644
--- 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,