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

Fix GPU-less tests.

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
test/cuda/cuda.jl
View File

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

2
test/runtests.jl
View File

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