Merge branch 'tests'

Author: ChrisRackauckas

.gitlab-ci.yml

@@ -0,0 +1,58 @@
+image: "julia:1"
+
+variables:
+  JULIA_DEPOT_PATH: "$CI_PROJECT_DIR/.julia/"
+  JULIA_NUM_THREADS: '8'
+
+cache:
+  paths:
+    - .julia/
+
+build:
+  stage: build
+  tags:
+    - 'p6000'
+  script:
+    - curl https://julialang-s3.julialang.org/bin/linux/x64/1.1/julia-1.1.1-linux-x86_64.tar.gz -o julia.tar.gz
+    - unp julia.tar.gz
+    - export PATH="$(pwd)/julia-1.1.1/bin:$PATH"
+    - julia -e "using InteractiveUtils;
+                versioninfo()"
+    - julia --project -e "using Pkg;
+                          Pkg.update();
+                          Pkg.instantiate();"
+  only:
+  - master
+  - tags
+  - external
+  - pushes
+  artifacts:
+    untracked: true
+    paths:
+      - .julia/**/*
+      - julia-1.1.1/**/*
+
+test-GPU:
+  stage: test
+  tags:
+    - 'p6000'
+  dependencies:
+    - build
+  variables:
+    GROUP: "GPU"
+  script:
+    - export PATH="$(pwd)/julia-1.1.1/bin:$PATH"
+    - julia -e "using InteractiveUtils;
+                versioninfo()"
+    - julia --project -e "using Pkg; Pkg.add(\"CuArrays\");
+                          Pkg.test(\"DiffEqFlux\"; coverage=true);"
+  only:
+  - master
+  - tags
+  - external
+  - pushes
+  artifacts:
+    untracked: true
+    paths:
+      - .julia/**/*
+      - julia-1.1.1/**/*

test/neural_de_gpu.jl

@@ -104,26 +104,28 @@ neural_ode_rd(dudt,u0,tspan,Tsit5(),saveat=0.1)
 # Adjoint
 
 @testset "adjoint mode trackedu0" begin
-    Tracker.zero_grad!(dudt[1].W.grad)
-    Tracker.zero_grad!(downsample.W.grad)
-    m1 = Chain(downsample, u0->neural_ode(dudt,u0,tspan,Tsit5(),save_everystep=false,save_start=false)) #broke
-    Flux.back!(sum(m1(x0)))
-    @test ! iszero(Tracker.grad(dudt[1].W))
-    @test ! iszero(Tracker.grad(downsample.W))
-
-    Tracker.zero_grad!(dudt[1].W.grad)
-    Tracker.zero_grad!(downsample.W.grad)
-    m2 = Chain(downsample, u0->neural_ode(dudt,u0,tspan,Tsit5(),saveat=0.0:0.1:10.0))
-    Flux.back!(sum(m2(x0)))
-    @test ! iszero(Tracker.grad(dudt[1].W))
-    @test ! iszero(Tracker.grad(downsample.W))
-
-    Tracker.zero_grad!(dudt[1].W.grad)
-    Tracker.zero_grad!(downsample.W.grad)
-    m3 = Chain(downsample, u0->neural_ode(dudt,u0,tspan,Tsit5(),saveat=0.1))
-    @test_broken Flux.back!(sum(m3(x0)))
-    #@test ! iszero(Tracker.grad(dudt[1].W))
-    #@test ! iszero(Tracker.grad(downsample.W))
+    @test_broken
+        Tracker.zero_grad!(dudt[1].W.grad)
+        Tracker.zero_grad!(downsample.W.grad)
+        m1 = Chain(downsample, u0->neural_ode(dudt,u0,tspan,Tsit5(),save_everystep=false,save_start=false)) #broke
+        Flux.back!(sum(m1(x0)))
+        @test ! iszero(Tracker.grad(dudt[1].W))
+        @test ! iszero(Tracker.grad(downsample.W))
+
+        Tracker.zero_grad!(dudt[1].W.grad)
+        Tracker.zero_grad!(downsample.W.grad)
+        m2 = Chain(downsample, u0->neural_ode(dudt,u0,tspan,Tsit5(),saveat=0.0:0.1:10.0))
+        Flux.back!(sum(m2(x0)))
+        @test ! iszero(Tracker.grad(dudt[1].W))
+        @test ! iszero(Tracker.grad(downsample.W))
+
+        Tracker.zero_grad!(dudt[1].W.grad)
+        Tracker.zero_grad!(downsample.W.grad)
+        m3 = Chain(downsample, u0->neural_ode(dudt,u0,tspan,Tsit5(),saveat=0.1))
+        @test_broken Flux.back!(sum(m3(x0)))
+        #@test ! iszero(Tracker.grad(dudt[1].W))
+        #@test ! iszero(Tracker.grad(downsample.W))
+    end
 end;
 
 #= # RD =#

test/runtests.jl

@@ -1,9 +1,22 @@
 using DiffEqFlux, Test, SafeTestsets
 
-@safetestset "Utils Tests" begin include("utils.jl") end
-@safetestset "Layers Tests" begin include("layers.jl") end
-@safetestset "Layers SDE" begin include("layers_sde.jl") end
-@safetestset "Layers DDE" begin include("layers_dde.jl") end
-@safetestset "odenet" begin include("odenet.jl") end
-@safetestset "Neural DE Tests" begin include("neural_de.jl") end
-@safetestset "Partial Neural Tests" begin include("partial_neural.jl") end
+const GROUP = get(ENV, "GROUP", "All")
+const is_APPVEYOR = ( Sys.iswindows() && haskey(ENV,"APPVEYOR") )
+const is_TRAVIS = haskey(ENV,"TRAVIS")
+
+@time begin
+if GROUP == "All"
+    @safetestset "Utils Tests" begin include("utils.jl") end
+    @safetestset "Layers Tests" begin include("layers.jl") end
+    @safetestset "Layers SDE" begin include("layers_sde.jl") end
+    @safetestset "Layers DDE" begin include("layers_dde.jl") end
+    @safetestset "odenet" begin include("odenet.jl") end
+    @safetestset "Neural DE Tests" begin include("neural_de.jl") end
+    @safetestset "Partial Neural Tests" begin include("partial_neural.jl") end
+end
+end
+
+if !is_APPVEYOR && GROUP == "GPU"
+  @safetestset "odenet GPU" begin include("odenet_gpu.jl") end
+  @safetestset "Neural DE GPU Tests" begin include("neural_de_gpu.jl") end
+end