Commit 26b433b5 by Tianqi Chen Committed by GitHub

[TEST] Jenkinsfile (#389)

* [TEST] Jenkinsfile

* Fix wheel setup
parent 90809381
...@@ -91,6 +91,7 @@ stage('Build') { ...@@ -91,6 +91,7 @@ stage('Build') {
echo LLVM_CONFIG=llvm-config-4.0 >> config.mk echo LLVM_CONFIG=llvm-config-4.0 >> config.mk
echo USE_RPC=1 >> config.mk echo USE_RPC=1 >> config.mk
echo USE_BLAS=openblas >> config.mk echo USE_BLAS=openblas >> config.mk
rm -f lib/libtvm_runtime.so lib/libtvm.so
""" """
make('gpu', '-j2') make('gpu', '-j2')
sh "mv lib/libtvm.so lib/libtvm_llvm40.so" sh "mv lib/libtvm.so lib/libtvm_llvm40.so"
......
...@@ -85,7 +85,7 @@ if "bdist_wheel" in sys.argv: ...@@ -85,7 +85,7 @@ if "bdist_wheel" in sys.argv:
with open("MANIFEST.in", "w") as fo: with open("MANIFEST.in", "w") as fo:
fo.write("include tvm/%s\n" % LIB_NAME) fo.write("include tvm/%s\n" % LIB_NAME)
setup_kwargs = { setup_kwargs = {
"include_package_data": False "include_package_data": True
} }
else: else:
setup_kwargs = { setup_kwargs = {
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment