upd

flashinfer-ai · Jan 28, 2024 · ea6fc19 · ea6fc19
1 parent 1426742
commit ea6fc19
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/python/MANIFEST.in b/python/MANIFEST.in
@@ -7,4 +7,5 @@ exclude flashinfer/_build_meta.py
 
 # Unneeded files
 prune */__pycache__
+tests/
 global-exclude *.so
diff --git a/python/setup.py b/python/setup.py
@@ -85,7 +85,7 @@ def remove_unwanted_pytorch_nvcc_flags():
                 "csrc/batch_prefill.cu",
             ],
             include_dirs=[
-                str("include"),
+                str(root / "include"),
             ],
             extra_compile_args={
                 "cxx": ["-O3"],