Skip to content

Commit

Permalink
CI Trigger
Browse files Browse the repository at this point in the history
  • Loading branch information
ArnoStrouwen committed Jan 21, 2024
1 parent 29e466e commit 8e87e8a
Show file tree
Hide file tree
Showing 7 changed files with 14 additions and 236 deletions.
79 changes: 0 additions & 79 deletions .buildkite/pipeline.yml

This file was deleted.

10 changes: 1 addition & 9 deletions .github/workflows/CI.yml
Original file line number Diff line number Diff line change
Expand Up @@ -12,20 +12,12 @@ on:
- 'docs/**'
jobs:
test:
runs-on: ubuntu-latest
runs-on: windows-latest
strategy:
fail-fast: false
matrix:
group:
- Core1
- Core2
- Core3
- Core4
- Core5
- Core6
- SDE1
- SDE2
- SDE3
version:
- '1.10'
steps:
Expand Down
55 changes: 0 additions & 55 deletions .github/workflows/Downstream.yml

This file was deleted.

42 changes: 0 additions & 42 deletions .github/workflows/FormatCheck.yml

This file was deleted.

40 changes: 0 additions & 40 deletions .github/workflows/Invalidations.yml

This file was deleted.

16 changes: 9 additions & 7 deletions test/adjoint_shapes.jl
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
using OrdinaryDiffEq, SciMLSensitivity, Zygote

tspan = (0.0, 1.0)
#= tspan = (0.0, 1.0)
X = randn(3, 4)
p = randn(3, 4)
f(u, p, t) = u .* p
Expand All @@ -10,7 +10,7 @@ Zygote.gradient(p -> sum(solve(prob_ube, Midpoint(), u0 = X, p = p)), p)
prob_ube = ODEProblem{true}(f, X, tspan, p)
Zygote.gradient(p -> sum(solve(prob_ube, Midpoint(), u0 = X, p = p)), p)

=#
function aug_dynamics!(dz, z, K, t)
x = @view z[2:end]
u = -K * x
Expand All @@ -20,14 +20,15 @@ end

policy_params = ones(2, 2)
z0 = zeros(3)
fwd_sol = solve(ODEProblem(aug_dynamics!, z0, (0.0, 1.0), policy_params),
#= fwd_sol = solve(ODEProblem(aug_dynamics!, z0, (0.0, 1.0), policy_params),
Tsit5(), u0 = z0, p = policy_params)

=#
test_prob = ODEProblem(aug_dynamics!, z0, (0.0, 1.0), policy_params)
sensealg = InterpolatingAdjoint()
sensealg = SciMLSensitivity.setvjp(sensealg,
SciMLSensitivity.inplace_vjp(fwd_sol.prob, fwd_sol.prob.u0, fwd_sol.prob.p, true))

solve(ODEAdjointProblem(fwd_sol, sensealg, Tsit5(),
SciMLSensitivity.inplace_vjp(test_prob, test_prob.u0, test_prob.p, true))
@test 1 == 1
#= solve(ODEAdjointProblem(fwd_sol, sensealg, Tsit5(),
[1.0], (out, x, p, t, i) -> (out .= 1)), Tsit5())
A = ones(2, 2)
Expand All @@ -49,3 +50,4 @@ fwd_sol = solve(ODEProblem(aug_dynamics!, z0, (0.0, 1.0), policy_params), u0 = z
solve(ODEAdjointProblem(fwd_sol, sensealg, Tsit5(), [1.0],
(out, x, p, t, i) -> (out .= 1)), Tsit5())
=#
8 changes: 4 additions & 4 deletions test/runtests.jl
Original file line number Diff line number Diff line change
Expand Up @@ -11,10 +11,10 @@ end

@time begin
if GROUP == "All" || GROUP == "Core1" || GROUP == "Downstream"
@time @safetestset "Forward Sensitivity" include("forward.jl")
@time @safetestset "Sparse Adjoint Sensitivity" include("sparse_adjoint.jl")
#= @time @safetestset "Forward Sensitivity" include("forward.jl")
@time @safetestset "Sparse Adjoint Sensitivity" include("sparse_adjoint.jl") =#
@time @safetestset "Adjoint Shapes" include("adjoint_shapes.jl")
@time @safetestset "Second Order Sensitivity" include("second_order.jl")
#= @time @safetestset "Second Order Sensitivity" include("second_order.jl")
@time @safetestset "Concrete Solve Derivatives" include("concrete_solve_derivatives.jl")
@time @safetestset "Branching Derivatives" include("branching_derivatives.jl")
@time @safetestset "Derivative Shapes" include("derivative_shapes.jl")
Expand All @@ -25,7 +25,7 @@ end
@time @safetestset "Forward Remake" include("forward_remake.jl")
@time @safetestset "Prob Kwargs" include("prob_kwargs.jl")
@time @safetestset "DiscreteProblem Adjoints" include("discrete.jl")
@time @safetestset "Time Type Mixing Adjoints" include("time_type_mixing.jl")
@time @safetestset "Time Type Mixing Adjoints" include("time_type_mixing.jl") =#
end

if GROUP == "All" || GROUP == "Core2"
Expand Down

0 comments on commit 8e87e8a

Please sign in to comment.