Fix tvm dependency for docker (apache#12479)

anirudh2290 · Sep 7, 2018 · 2bcfc08 · 2bcfc08
1 parent 285adda
commit 2bcfc08
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 8 deletions.
diff --git 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
@@ -21,7 +21,6 @@
 import numpy as np
 import unittest
 
-@unittest.skip("Flaky test https://github.com/apache/incubator-mxnet/issues/12473")
 def test_tvm_bridge():
     # only enable test if TVM is available
     try: