Merge pull request #422 from tejank10/cudnn_avail

cudnn_available update
2018-10-05 12:05:18 +01:00 · 2018-10-05 12:05:18 +01:00 · f08b6f80d2
commit f08b6f80d2
parent 683bbec71c 2ff54ee0fd
2 changed files with 2 additions and 2 deletions
--- a/src/cuda/cuda.jl
+++ b/src/cuda/cuda.jl
@ -2,6 +2,6 @@ module CUDA

 using ..CuArrays

-CuArrays.cudnn_available() && include("cudnn.jl")
+CuArrays.libcudnn != nothing && include("cudnn.jl")

 end
--- a/test/cuda/cuda.jl
+++ b/test/cuda/cuda.jl
@ -36,4 +36,4 @@ Flux.back!(sum(l))

 end

-CuArrays.cudnn_available() && include("cudnn.jl")
+CuArrays.libcudnn != nothing && include("cudnn.jl")