Skip to content

Commit bbeb0ce

Browse files
fix: fix nothing pullback for parameter object in Zygote adjoint
1 parent cc43377 commit bbeb0ce

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

ext/SciMLBaseZygoteExt.jl

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -48,6 +48,9 @@ end
4848
grz = pullback(getter, sym, VA.u[j], VA.prob.p, VA.t[j])[2](Δ)
4949
du = [k == j ? grz[2] : zero(VA.u[1]) for k in 1:length(VA.u)]
5050
dp = grz[3] # pullback for p
51+
if dp === nothing
52+
dp = parameter_values(VA)
53+
end
5154
dprob = remake(VA.prob, p = dp)
5255
du, dprob
5356
else

0 commit comments

Comments
 (0)