From 18b95458a09c4b3dda4bb0983ccdc759fbf40311 Mon Sep 17 00:00:00 2001
From: Yizhi Liu <liuyizhi@apache.org>
Date: Mon, 2 Sep 2019 19:02:22 -0700
Subject: [PATCH] hack: add cuda to ld_path in cmake

---
 CMakeLists.txt                 | 2 +-
 ci/docker/runtime_functions.sh | 4 ++++
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index 33529747f44c..ef73a74c1736 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -761,7 +761,7 @@ if(USE_TVM_OP)
   add_custom_command(TARGET mxnet POST_BUILD
     COMMAND ${CMAKE_COMMAND} -E env
           PYTHONPATH="${CMAKE_CURRENT_SOURCE_DIR}/3rdparty/tvm/python:${CMAKE_CURRENT_SOURCE_DIR}/3rdparty/tvm/topi/python:${CMAKE_CURRENT_SOURCE_DIR}/contrib"
-          LD_LIBRARY_PATH="${CMAKE_CURRENT_BINARY_DIR}/3rdparty/tvm"
+          LD_LIBRARY_PATH="${CMAKE_CURRENT_BINARY_DIR}/3rdparty/tvm:/usr/local/cuda/lib64:/usr/local/cuda/targets/x86_64-linux/lib/stubs"
           ${Python3_EXECUTABLE} ${CMAKE_CURRENT_SOURCE_DIR}/contrib/tvmop/compile.py -o${CMAKE_CURRENT_BINARY_DIR}/libtvmop.so
     )
 endif()
diff --git a/ci/docker/runtime_functions.sh b/ci/docker/runtime_functions.sh
index a60fec07246a..b46788452b07 100755
--- a/ci/docker/runtime_functions.sh
+++ b/ci/docker/runtime_functions.sh
@@ -741,6 +741,9 @@ build_ubuntu_gpu_cmake() {
     set -ex
     cd /work/build
     build_ccache_wrappers
+    echo "ls cuda path"
+    ls /usr/local/cuda/lib64
+    ls /usr/local/cuda/targets/x86_64-linux/lib/stubs
     cmake \
         -DCMAKE_CXX_COMPILER_LAUNCHER=ccache    \
         -DCMAKE_C_COMPILER_LAUNCHER=ccache      \
@@ -798,6 +801,7 @@ build_ubuntu_gpu_large_tensor() {
         -DUSE_CUDA=ON                           \
         -DUSE_CUDNN=ON                          \
         -DUSE_TVM_OP=ON                         \
+        -DPython3_EXECUTABLE=/usr/bin/python3   \
         -DUSE_MKL_IF_AVAILABLE=OFF              \
         -DUSE_MKLML_MKL=OFF                     \
         -DUSE_MKLDNN=OFF                        \