Skip to content

Commit b4d8029

Browse files
committed
fix: return the result always
1 parent 229b69d commit b4d8029

File tree

1 file changed

+5
-6
lines changed

1 file changed

+5
-6
lines changed

lib/SciMLJacobianOperators/src/SciMLJacobianOperators.jl

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -150,18 +150,17 @@ function (op::JacobianOperator)(Jv, v, u, p)
150150
if op.mode isa VJP
151151
if SciMLBase.isinplace(op)
152152
op.vjp_op(Jv, v, u, p)
153-
return
153+
else
154+
copyto!(Jv, op.vjp_op(v, u, p))
154155
end
155-
copyto!(Jv, op.vjp_op(v, u, p))
156-
return
157156
else
158157
if SciMLBase.isinplace(op)
159158
op.jvp_op(Jv, v, u, p)
160-
return
159+
else
160+
copyto!(Jv, op.jvp_op(v, u, p))
161161
end
162-
copyto!(Jv, op.jvp_op(v, u, p))
163-
return
164162
end
163+
return Jv
165164
end
166165

167166
"""

0 commit comments

Comments
 (0)