From 7f1b0f5baef2a0fd47857b6096e6239b4ef85dc8 Mon Sep 17 00:00:00 2001
From: Tianqi Chen <tqchen@users.noreply.github.com>
Date: Thu, 24 Aug 2017 09:00:03 -0700
Subject: [PATCH] [SETUP] Fix python setup (#380)
---
python/setup.py | 21 ++++++++++++++-------
1 file changed, 14 insertions(+), 7 deletions(-)
diff --git a/python/setup.py b/python/setup.py
index 883278c11..38ad37a3b 100644
--- a/python/setup.py
+++ b/python/setup.py
@@ -82,25 +82,32 @@ class BinaryDistribution(Distribution):
# For bdist_wheel only
if "bdist_wheel" in sys.argv:
shutil.copy(LIB_PATH[0], os.path.join(CURRENT_DIR, 'tvm'))
- fo = open("MANIFEST.in", "w")
- fo.write("include tvm/%s\n" % LIB_NAME)
- fo.close()
+ with open("MANIFEST.in", "w") as fo:
+ fo.write("include tvm/%s\n" % LIB_NAME)
+ setup_kwargs = {
+ "include_package_data": False
+ }
+else:
+ setup_kwargs = {
+ "include_package_data": True,
+ "data_files": [('tvm', [LIB_PATH[0]])]
+ }
setup(name='tvm',
version=__version__,
- description='A domain specific language(DSL) for tensor computations.',
+ description="TVM: An End to End Tensor IR/DSL Stack for Deep Learning Systems",
zip_safe=False,
install_requires=[
'numpy',
'decorator',
],
packages=find_packages(),
- include_package_data=True,
distclass=BinaryDistribution,
url='https://github.com/dmlc/tvm',
- ext_modules=config_cython())
+ ext_modules=config_cython(),
+ **setup_kwargs)
-# clean up
+# Wheel cleanup
if "bdist_wheel" in sys.argv:
os.remove("MANIFEST.in")
os.remove("tvm/%s" % LIB_NAME)
--
GitLab