Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Fix tvm dependency for docker #12479

Merged
merged 1 commit into from
Sep 7, 2018
+7 −8
Merged

Fix tvm dependency for docker #12479

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
14 changes: 7 additions & 7 deletions ci/docker/install/ubuntu_tvm.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
1 change: 0 additions & 1 deletion tests/python/gpu/test_tvm_bridge.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down