conditionally execute RNN tests
This commit is contained in:
parent
bc33108e66
commit
cff1dfd258
|
@ -1,3 +1,6 @@
|
||||||
|
before_script:
|
||||||
|
- export CI_CURNN_TEST=true
|
||||||
|
|
||||||
variables:
|
variables:
|
||||||
CI_IMAGE_TAG: 'cuda'
|
CI_IMAGE_TAG: 'cuda'
|
||||||
|
|
||||||
|
|
|
@ -41,5 +41,7 @@ end
|
||||||
if CuArrays.libcudnn != nothing
|
if CuArrays.libcudnn != nothing
|
||||||
@info "Testing Flux/CUDNN"
|
@info "Testing Flux/CUDNN"
|
||||||
include("cudnn.jl")
|
include("cudnn.jl")
|
||||||
# include("curnn.jl")
|
if !haskey(ENV, "CI_CURNN_TEST")
|
||||||
|
include("curnn.jl")
|
||||||
|
end
|
||||||
end
|
end
|
||||||
|
|
Loading…
Reference in New Issue