diff --git a/lib/OrdinaryDiffEqDifferentiation/src/alg_utils.jl b/lib/OrdinaryDiffEqDifferentiation/src/alg_utils.jl index 8d767433fd..76161503c2 100644 --- a/lib/OrdinaryDiffEqDifferentiation/src/alg_utils.jl +++ b/lib/OrdinaryDiffEqDifferentiation/src/alg_utils.jl @@ -53,7 +53,7 @@ function DiffEqBase.prepare_alg( if alg_autodiff(alg) isa AutoForwardDiff && ((prob.f isa ODEFunction && prob.f.f isa FunctionWrappersWrappers.FunctionWrappersWrapper) || (isbitstype(T) && sizeof(T) > 24)) - return remake(alg, autodiff = AutoForwardDiff(chunksize = 1, tag = _get_fwd_tag(alg_autodiff(alg)))) + return remake(alg, autodiff = AutoForwardDiff(chunksize = 1, tag = alg_autodiff(alg).tag)) end # If the autodiff alg is AutoFiniteDiff, prob.f.f isa FunctionWrappersWrapper, diff --git a/lib/OrdinaryDiffEqNonlinearSolve/src/OrdinaryDiffEqNonlinearSolve.jl b/lib/OrdinaryDiffEqNonlinearSolve/src/OrdinaryDiffEqNonlinearSolve.jl index 461ece35cb..3453c6fd73 100644 --- a/lib/OrdinaryDiffEqNonlinearSolve/src/OrdinaryDiffEqNonlinearSolve.jl +++ b/lib/OrdinaryDiffEqNonlinearSolve/src/OrdinaryDiffEqNonlinearSolve.jl @@ -25,7 +25,7 @@ import SciMLStructures: canonicalize, Tunable, isscimlstructure import OrdinaryDiffEqCore import SciMLOperators: islinear -import OrdinaryDiffEqCore: nlsolve_f, set_new_W!, set_W_γdt!, _ADType_to_Bool +import OrdinaryDiffEqCore: nlsolve_f, set_new_W!, set_W_γdt! @static if isdefined(OrdinaryDiffEqCore, :default_nlsolve) import OrdinaryDiffEqCore: default_nlsolve