diff --git a/setup.py b/setup.py index f6f3fb171..46c95e454 100644 --- a/setup.py +++ b/setup.py @@ -226,29 +226,14 @@ def run(self): # Copy the built TVM to the package directory TVM_PREBUILD_ITEMS = [ - "3rdparty/tvm/build/libtvm_runtime.so", - "3rdparty/tvm/build/libtvm.so", - "3rdparty/tvm/build/config.cmake", - "3rdparty/tvm/python", - "3rdparty/tvm/licenses", - "3rdparty/tvm/conftest.py", - "3rdparty/tvm/CONTRIBUTORS.md", - "3rdparty/tvm/KEYS", - "3rdparty/tvm/LICENSE", - "3rdparty/tvm/README.md", - "3rdparty/tvm/mypy.ini", - "3rdparty/tvm/pyproject.toml", - "3rdparty/tvm/version.py", - "3rdparty/tvm/common.h", - "3rdparty/tvm/copy.h", - "3rdparty/tvm/copy_sm90.h", - "3rdparty/tvm/gemm.h", - "3rdparty/tvm/gemm_sm70.h", - "3rdparty/tvm/gemm_sm80.h", - "3rdparty/tvm/gemm_sm90.h", - "3rdparty/tvm/ldsm.h", - "3rdparty/tvm/reduce.h", - "3rdparty/tvm/threadblock_swizzle.h" + "3rdparty/tvm/build/libtvm_runtime.so", "3rdparty/tvm/build/libtvm.so", + "3rdparty/tvm/build/config.cmake", "3rdparty/tvm/python", "3rdparty/tvm/licenses", + "3rdparty/tvm/conftest.py", "3rdparty/tvm/CONTRIBUTORS.md", "3rdparty/tvm/KEYS", + "3rdparty/tvm/LICENSE", "3rdparty/tvm/README.md", "3rdparty/tvm/mypy.ini", + "3rdparty/tvm/pyproject.toml", "3rdparty/tvm/version.py", "3rdparty/tvm/common.h", + "3rdparty/tvm/copy.h", "3rdparty/tvm/copy_sm90.h", "3rdparty/tvm/gemm.h", + "3rdparty/tvm/gemm_sm70.h", "3rdparty/tvm/gemm_sm80.h", "3rdparty/tvm/gemm_sm90.h", + "3rdparty/tvm/ldsm.h", "3rdparty/tvm/reduce.h", "3rdparty/tvm/threadblock_swizzle.h" ] for item in TVM_PREBUILD_ITEMS: source_dir = os.path.join(ROOT_DIR, item)