Merge pull request #422 from tejank10/cudnn_avail

cudnn_available update
This commit is contained in:
Mike J Innes 2018-10-05 12:05:18 +01:00 committed by GitHub
commit f08b6f80d2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions

View File

@ -2,6 +2,6 @@ module CUDA
using ..CuArrays
CuArrays.cudnn_available() && include("cudnn.jl")
CuArrays.libcudnn != nothing && include("cudnn.jl")
end

View File

@ -36,4 +36,4 @@ Flux.back!(sum(l))
end
CuArrays.cudnn_available() && include("cudnn.jl")
CuArrays.libcudnn != nothing && include("cudnn.jl")