Skip to content

Commit 977639a

Browse files
Merge pull request #1168 from SciML/mooncake_origina
Make sure to set MooncakeOriginator
2 parents f86c4ee + 492bb9d commit 977639a

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)