Skip to content

Commit c3ca370

Browse files
committed
change to Annotation because sometimes we get DuplicatedNoNeed
1 parent f4f04f4 commit c3ca370

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

lib/OptimizationBase/ext/OptimizationEnzymeExt.jl

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -830,7 +830,6 @@ Enzyme.EnzymeRules.inactive_kwarg(::typeof(OptimizationBase.solve_up), prob, sen
830830

831831
Enzyme.EnzymeRules.has_easy_rule(::typeof(OptimizationBase.solve_up), prob, sensalg::Union{Nothing,SciMLBase.AbstractSensitivityAlgorithm}, u0, p, args...; kwargs...) = nothing
832832

833-
834833
function Enzyme.EnzymeRules.augmented_primal(
835834
config::Enzyme.EnzymeRules.RevConfigWidth{1},
836835
func::Const{typeof(OptimizationBase.solve_up)}, RTA::Type{Duplicated{RT}}, prob,
@@ -868,7 +867,7 @@ end
868867

869868
function Enzyme.EnzymeRules.reverse(
870869
config::Enzyme.EnzymeRules.RevConfigWidth{1},
871-
func::Const{typeof(OptimizationBase.solve_up)}, ::Type{Duplicated{RT}}, tape, prob,
870+
func::Const{typeof(OptimizationBase.solve_up)}, ::Type{<:Enzyme.Annotation{RT}}, tape, prob,
872871
sensealg::Union{
873872
Const{Nothing}, Const{<:SciMLBase.AbstractSensitivityAlgorithm},
874873
},

0 commit comments

Comments
 (0)