Skip to content

Commit 492bb9d

Browse files
Make sure to set MooncakeOriginator
Should fix up downstream SciMLSensitivity Core 5
1 parent f86c4ee commit 492bb9d

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

src/solve.jl

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1093,9 +1093,13 @@ function solve(prob::AbstractDEProblem, args...; sensealg = nothing,
10931093
p = p !== nothing ? p : prob.p
10941094

10951095
if wrap isa Val{true}
1096-
wrap_sol(solve_up(prob, sensealg, u0, p, args...; kwargs...))
1096+
wrap_sol(solve_up(prob, sensealg, u0, p, args...;
1097+
originator = set_mooncakeoriginator_if_mooncake(SciMLBase.ChainRulesOriginator()),
1098+
kwargs...))
10971099
else
1098-
solve_up(prob, sensealg, u0, p, args...; kwargs...)
1100+
solve_up(prob, sensealg, u0, p, args...;
1101+
originator = set_mooncakeoriginator_if_mooncake(SciMLBase.ChainRulesOriginator()),
1102+
kwargs...)
10991103
end
11001104
end
11011105

0 commit comments

Comments
 (0)