From 6bd3951d93999d1d58681ca56b2269bb4a7558a4 Mon Sep 17 00:00:00 2001
From: Valentin Churavy <v.churavy@gmail.com>
Date: Fri, 16 Sep 2016 03:42:17 +0900
Subject: [PATCH] try to automatically find CUDA

---
 deps/build.jl | 17 ++++++++++++++++-
 1 file changed, 16 insertions(+), 1 deletion(-)

diff --git a/deps/build.jl b/deps/build.jl
index b79940e8c1eb..4a13b19c105c 100644
--- a/deps/build.jl
+++ b/deps/build.jl
@@ -32,6 +32,16 @@ if !libmxnet_detected
 
   blas_path = Libdl.dlpath(Libdl.dlopen(Base.libblas_name))
 
+  # Try to find cuda
+  hascuda = false
+  if haskey(ENV, "CUDA_HOME")
+    hascuda = Libdl.dlopen_e(joinpath(ENV["CUDA_HOME"], "lib64", "libcuda.so")) != C_NULL
+  else
+    cudapaths = String["/opt/cuda/lib64", "/usr/local/cuda/lib64"]
+    cudalib = Libdl.find_library(["libcuda", "libcuda.so"], cudapaths)
+    hascuda = Libdl.dlopen_e(cudalib) != C_NULL
+  end
+
   if VERSION >= v"0.5.0-dev+4338"
     blas_vendor = Base.BLAS.vendor()
   else
@@ -80,7 +90,6 @@ if !libmxnet_detected
         end)
         @build_steps begin
           ChangeDirectory(_mxdir)
-          # TODO(vchuravy). We have to reset mshadow/make/mshadow.mk
           `git -C mshadow checkout -- make/mshadow.mk`
           `git fetch`
           `git checkout $libmxnet_curr_ver`
@@ -95,6 +104,12 @@ if !libmxnet_detected
             `cp make/config.mk config.mk`
           end
           `sed -i -s 's/USE_OPENCV = 1/USE_OPENCV = 0/' config.mk`
+          if hascuda
+            `sed -i -s 's/USE_CUDA = 0/USE_CUDA = 1/' config.mk`
+            if haskey(ENV, "CUDA_HOME")
+              `sed -i -s 's/USE_CUDA_PATH = NULL/USE_CUDA_PATH = $(ENV["CUDA_HOME"])/' config.mk`
+            end
+          end
         end)
         @build_steps begin
           ChangeDirectory(_mxdir)