|
30 | 30 | du01, dp1 = Zygote.gradient(sum_of_solution, u0, p) |
31 | 31 |
|
32 | 32 | # These tests depend on a ZygoteRule in a package extension |
33 | | -# package exentsions do not exist before 1.9, so they cannot work. |
34 | | -if VERSION >= v"1.9" |
35 | | - function symbolic_indexing(u0, p) |
36 | | - _prob = remake(prob, u0 = u0, p = p) |
37 | | - soln = solve(_prob, Tsit5(), reltol = 1e-6, abstol = 1e-6, saveat = 0.1, |
38 | | - sensealg = BacksolveAdjoint(autojacvec = ZygoteVJP())) |
39 | | - sum(soln[x]) |
40 | | - end |
41 | | - |
42 | | - du01, dp1 = Zygote.gradient(symbolic_indexing, u0, p) |
43 | | - |
44 | | - function symbolic_indexing_observed(u0, p) |
45 | | - _prob = remake(prob, u0 = u0, p = p) |
46 | | - soln = solve(_prob, Tsit5(), reltol = 1e-6, abstol = 1e-6, saveat = 0.1, |
47 | | - sensealg = BacksolveAdjoint(autojacvec = ZygoteVJP())) |
48 | | - sum(soln[o, i] for i in 1:length(soln)) |
49 | | - end |
50 | | - |
51 | | - du01, dp1 = Zygote.gradient(symbolic_indexing_observed, u0, p) |
| 33 | +function symbolic_indexing(u0, p) |
| 34 | + _prob = remake(prob, u0 = u0, p = p) |
| 35 | + soln = solve(_prob, Tsit5(), reltol = 1e-6, abstol = 1e-6, saveat = 0.1, |
| 36 | + sensealg = BacksolveAdjoint(autojacvec = ZygoteVJP())) |
| 37 | + sum(soln[x]) |
| 38 | +end |
| 39 | + |
| 40 | +du01, dp1 = Zygote.gradient(symbolic_indexing, u0, p) |
| 41 | + |
| 42 | +function symbolic_indexing_observed(u0, p) |
| 43 | + _prob = remake(prob, u0 = u0, p = p) |
| 44 | + soln = solve(_prob, Tsit5(), reltol = 1e-6, abstol = 1e-6, saveat = 0.1, |
| 45 | + sensealg = BacksolveAdjoint(autojacvec = ZygoteVJP())) |
| 46 | + sum(soln[o, i] for i in 1:length(soln)) |
52 | 47 | end |
| 48 | + |
| 49 | +du01, dp1 = Zygote.gradient(symbolic_indexing_observed, u0, p) |
0 commit comments