diff --git a/CMakeLists.txt b/CMakeLists.txt index 4e96a7b15986..b90fc43e056c 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -694,7 +694,7 @@ if(NOT DEFINED ENV{CONDA_BUILD}) message(STATUS ${CMAKE_CURRENT_BINARY_DIR}) add_custom_target( tvm_cython ALL - ${Python_EXECUTABLE} setup.py build_ext --inplace + ${Python_EXECUTABLE} -I setup.py build_ext --inplace WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}/python ) add_dependencies(tvm_cython tvm) diff --git a/python/setup.py b/python/setup.py index 679f5078d3c1..cf2eff2a3af4 100644 --- a/python/setup.py +++ b/python/setup.py @@ -20,7 +20,6 @@ import pathlib import shutil import sys -import sys from setuptools import find_packages from setuptools.dist import Distribution