fix

flashinfer-ai · Jan 28, 2024 · c3e26dc · c3e26dc
1 parent 3572068
commit c3e26dc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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"],