This is an automated email from the ASF dual-hosted git repository.
tqchen pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/main by this push:
new a9fcac1a47 [Python] Fix setup.py for inplace build (#16214)
a9fcac1a47 is described below
commit a9fcac1a47f4b57c1d9d828c7ed1d77452b02cc5
Author: Ruihang Lai <[email protected]>
AuthorDate: Sun Dec 10 08:57:28 2023 -0500
[Python] Fix setup.py for inplace build (#16214)
Previously, `python/setup.py` missed to take the inplace build
case into account when doing file operations (such as finding
paths or copying files), which makes command `make cython` fail.
This PR fixes the issue by checking inplace builds.
---
python/setup.py | 7 ++++---
1 file changed, 4 insertions(+), 3 deletions(-)
diff --git a/python/setup.py b/python/setup.py
index 6901fbf864..594ab5fc8d 100644
--- a/python/setup.py
+++ b/python/setup.py
@@ -36,6 +36,7 @@ else:
CURRENT_DIR = os.path.dirname(__file__)
FFI_MODE = os.environ.get("TVM_FFI", "auto")
CONDA_BUILD = os.getenv("CONDA_BUILD") is not None
+INPLACE_BUILD = "--inplace" in sys.argv
def get_lib_path():
@@ -46,7 +47,7 @@ def get_lib_path():
libinfo = {"__file__": libinfo_py}
exec(compile(open(libinfo_py, "rb").read(), libinfo_py, "exec"), libinfo,
libinfo)
version = libinfo["__version__"]
- if not CONDA_BUILD:
+ if not CONDA_BUILD and not INPLACE_BUILD:
lib_path = libinfo["find_lib_path"]()
libs = [lib_path[0]]
if "runtime" not in libs[0]:
@@ -214,7 +215,7 @@ class BinaryDistribution(Distribution):
setup_kwargs = {}
-if not CONDA_BUILD:
+if not CONDA_BUILD and not INPLACE_BUILD:
with open("MANIFEST.in", "w") as fo:
for path in LIB_LIST:
if os.path.isfile(path):
@@ -286,7 +287,7 @@ setup(
)
-if not CONDA_BUILD:
+if not CONDA_BUILD and not INPLACE_BUILD:
# Wheel cleanup
os.remove("MANIFEST.in")
for path in LIB_LIST:
