From c3e26dc627676dc6c427e703654fa6e756b7a5a2 Mon Sep 17 00:00:00 2001
From: Zihao Ye <expye@outlook.com>
Date: Sun, 28 Jan 2024 23:58:58 +0800
Subject: [PATCH] fix

---
 python/setup.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/python/setup.py b/python/setup.py
index c26b6f13..b32ad916 100644
--- a/python/setup.py
+++ b/python/setup.py
@@ -20,7 +20,7 @@
 import setuptools
 import torch.utils.cpp_extension as torch_cpp_ext
 
-root = pathlib.Path(__name__).parent
+root = pathlib.Path(__name__).parent.resolve().parent
 
 
 def get_local_version_suffix() -> str:
@@ -73,7 +73,7 @@ def remove_unwanted_pytorch_nvcc_flags():
             "csrc/batch_prefill.cu",
         ],
         include_dirs=[
-            str(root.resolve().parent / "include"),
+            str(root / "include"),
         ],
         extra_compile_args={
             "cxx": ["-O3"],