Merge #712
712: Enable GPU CI r=dhairyagandhi96 a=dhairyagandhi96 Looking for feedback on this policy for doing GPU CI. Co-authored-by: Dhairya Gandhi <dhairya@juliacopmuting.com>
This commit is contained in:
commit
25097c4322
|
@ -1,10 +1,37 @@
|
|||
before_script:
|
||||
- export CI_DISABLE_CURNN_TEST=true
|
||||
|
||||
variables:
|
||||
CI_IMAGE_TAG: 'cuda'
|
||||
|
||||
include:
|
||||
- 'https://raw.githubusercontent.com/JuliaGPU/gitlab-ci/master/templates/v1/common.yml'
|
||||
- 'https://raw.githubusercontent.com/JuliaGPU/gitlab-ci/master/templates/v1/test_v1.0.yml'
|
||||
- 'https://raw.githubusercontent.com/JuliaGPU/gitlab-ci/master/templates/v1/test_dev.yml'
|
||||
- 'https://raw.githubusercontent.com/JuliaGPU/gitlab-ci/master/templates/v3/common.yml'
|
||||
|
||||
test:dev:
|
||||
allow_failure: true
|
||||
.flux:
|
||||
extends: .test
|
||||
script:
|
||||
- julia -e 'using InteractiveUtils;
|
||||
versioninfo()'
|
||||
- mkdir $JULIA_DEPOT_PATH # Pkg3.jl#325
|
||||
- julia -e 'using Pkg;
|
||||
Pkg.add("CuArrays");'
|
||||
- julia --project -e 'using Pkg;
|
||||
Pkg.instantiate();
|
||||
Pkg.build();
|
||||
Pkg.test(; coverage=true);'
|
||||
|
||||
test:v1.0:
|
||||
extends: .flux
|
||||
variables:
|
||||
CI_VERSION_TAG: 'v1.0'
|
||||
only:
|
||||
- staging
|
||||
- trying
|
||||
|
||||
test:v1.1:
|
||||
extends: .flux
|
||||
variables:
|
||||
CI_VERSION_TAG: 'v1.1'
|
||||
only:
|
||||
- staging
|
||||
- trying
|
||||
|
|
|
@ -41,5 +41,7 @@ end
|
|||
if CuArrays.libcudnn != nothing
|
||||
@info "Testing Flux/CUDNN"
|
||||
include("cudnn.jl")
|
||||
include("curnn.jl")
|
||||
if !haskey(ENV, "CI_DISABLE_CURNN_TEST")
|
||||
include("curnn.jl")
|
||||
end
|
||||
end
|
||||
|
|
Loading…
Reference in New Issue