Fix GPU-less tests.
This commit is contained in:
parent
dbcdf4d1bd
commit
33d276cdb7
|
@ -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
|
||||
|
|
|
@ -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"
|
||||
|
|
Loading…
Reference in New Issue