Fix GPU-less tests.

This commit is contained in:
Tim Besard 2019-11-04 15:37:25 +01:00
parent dbcdf4d1bd
commit 33d276cdb7
2 changed files with 8 additions and 4 deletions

View File

@ -53,6 +53,10 @@ end
@test y[3,:] isa CuArray
end
@info "Testing Flux/CUDNN"
include("cudnn.jl")
include("curnn.jl")
if CuArrays.has_cudnn()
@info "Testing Flux/CUDNN"
include("cudnn.jl")
include("curnn.jl")
else
@warn "CUDNN unavailable, not testing GPU DNN support"
end

View File

@ -19,7 +19,7 @@ include("layers/normalisation.jl")
include("layers/stateless.jl")
include("layers/conv.jl")
if isdefined(Flux, :CUDA)
if Flux.use_cuda[]
include("cuda/cuda.jl")
else
@warn "CUDA unavailable, not testing GPU support"