|
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