TastyPancakes/Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

libcudnn check fix

Browse Source
This commit is contained in:
Mike J Innes 2019-08-27 15:41:23 +01:00
parent 9cd97f06f7
commit 61a8cfd6ee
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
test/cuda/cuda.jl
View File

@ -48,7 +48,7 @@ end
@test y[3,:] isa CuArray @test y[3,:] isa CuArray
end end
if has_cudnn() != nothing if CuArrays.libcudnn != nothing
@info "Testing Flux/CUDNN" @info "Testing Flux/CUDNN"
include("cudnn.jl") include("cudnn.jl")
if !haskey(ENV, "CI_DISABLE_CURNN_TEST") if !haskey(ENV, "CI_DISABLE_CURNN_TEST")