eric-haibin-lin closed pull request #12479: Fix tvm dependency for docker URL: https://github.com/apache/incubator-mxnet/pull/12479
This is a PR merged from a forked repository. As GitHub hides the original diff on merge, it is displayed below for the sake of provenance: As this is a foreign pull request (from a fork), the diff is supplied below (as it won't show otherwise due to GitHub magic): diff --git a/ci/docker/install/ubuntu_tvm.sh b/ci/docker/install/ubuntu_tvm.sh index 4f5cb4251ad..2ee4e534ed9 100755 --- a/ci/docker/install/ubuntu_tvm.sh +++ b/ci/docker/install/ubuntu_tvm.sh @@ -25,14 +25,14 @@ cd tvm # This is a stable tag that support MXNet TVM bridge. # We use this since support for mxnet bridge just checked # into master and there is yet a version tag -git checkout 30eaf463e34d7c301357c31a010945d11df16537 +git checkout v0.4 + +cp cmake/config.cmake . +echo set\(USE_CUDA /usr/local/cuda\) >> config.cmake +echo set\(USE_LLVM llvm-config-5.0\) >> config.cmake +echo set\(USE_RPC ON\) >> config.cmake +echo set\(USE_GRAPH_RUNTIME ON\) >> config.cmake -cp make/config.mk -echo USE_CUDA=1 >> config.mk -echo LLVM_CONFIG=llvm-config-5.0 >> config.mk -echo USE_RPC=1 >> config.mk -echo USE_GRAPH_RUNTIME=1 >> config.mk -echo CUDA_PATH=/usr/local/cuda >> config.mk make -j$(nproc) cd python diff --git a/tests/python/gpu/test_tvm_bridge.py b/tests/python/gpu/test_tvm_bridge.py index c3cf2ddba29..5c87536bdba 100644 --- a/tests/python/gpu/test_tvm_bridge.py +++ b/tests/python/gpu/test_tvm_bridge.py @@ -21,7 +21,6 @@ import numpy as np import unittest [email protected]("Flaky test https://github.com/apache/incubator-mxnet/issues/12473";) def test_tvm_bridge(): # only enable test if TVM is available try: ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: [email protected] With regards, Apache Git Services
