cudnn_available() update

2018-10-04 11:31:29 -04:00 · 2018-10-04 11:31:29 -04:00 · 2ff54ee0fd
parent 683bbec71c
commit 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")