From 492bb9d63da56b8bf706af351fd27c37322a4161 Mon Sep 17 00:00:00 2001 From: Christopher Rackauckas Date: Thu, 3 Jul 2025 11:11:07 +0000 Subject: [PATCH] Make sure to set MooncakeOriginator Should fix up downstream SciMLSensitivity Core 5 --- src/solve.jl | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/src/solve.jl b/src/solve.jl index 633e0246d..235f692d6 100644 --- a/src/solve.jl +++ b/src/solve.jl @@ -1093,9 +1093,13 @@ function solve(prob::AbstractDEProblem, args...; sensealg = nothing, p = p !== nothing ? p : prob.p if wrap isa Val{true} - wrap_sol(solve_up(prob, sensealg, u0, p, args...; kwargs...)) + wrap_sol(solve_up(prob, sensealg, u0, p, args...; + originator = set_mooncakeoriginator_if_mooncake(SciMLBase.ChainRulesOriginator()), + kwargs...)) else - solve_up(prob, sensealg, u0, p, args...; kwargs...) + solve_up(prob, sensealg, u0, p, args...; + originator = set_mooncakeoriginator_if_mooncake(SciMLBase.ChainRulesOriginator()), + kwargs...) end end