Skip to content

Commit 4f72bb1

Browse files
Fix extension dispatch
1 parent 3770c1c commit 4f72bb1

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

ext/DiffEqBaseEnzymeExt.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ isdefined(Base, :get_extension) ? (import Enzyme) : (import ..Enzyme)
77
using ChainRulesCore
88
using EnzymeCore
99

10-
function EnzymeCore.EnzymeRules.augmented_primal(config::EnzymeCore.EnzymeRules.ConfigWidth{1}, func::Const{typeof(DiffEqBase.solve_up)}, ::Type{Duplicated{RT}}, prob, sensealg::Union{Const{Nothing}, Const{<:AbstractSensitivityAlgorithm}}, u0, p, args...; kwargs...) where RT
10+
function EnzymeCore.EnzymeRules.augmented_primal(config::EnzymeCore.EnzymeRules.ConfigWidth{1}, func::Const{typeof(DiffEqBase.solve_up)}, ::Type{Duplicated{RT}}, prob, sensealg::Union{Const{Nothing}, Const{<:DiffEqBase.AbstractSensitivityAlgorithm}}, u0, p, args...; kwargs...) where RT
1111
@inline function copy_or_reuse(val, idx)
1212
if EnzymeCore.EnzymeRules.overwritten(config)[idx] && ismutable(val)
1313
return deepcopy(val)
@@ -31,7 +31,7 @@ function EnzymeCore.EnzymeRules.augmented_primal(config::EnzymeCore.EnzymeRules.
3131
return EnzymeCore.EnzymeRules.AugmentedReturn{RT, RT, Any}(res[1], dres, tup::Any)
3232
end
3333

34-
function EnzymeCore.EnzymeRules.reverse(config::EnzymeCore.EnzymeRules.ConfigWidth{1}, func::Const{typeof(DiffEqBase.solve_up)}, ::Type{<:Duplicated{RT}}, tape, prob, sensealg::Union{Const{Nothing}, Const{<:AbstractSensitivityAlgorithm}}, u0, p, args...; kwargs...) where RT
34+
function EnzymeCore.EnzymeRules.reverse(config::EnzymeCore.EnzymeRules.ConfigWidth{1}, func::Const{typeof(DiffEqBase.solve_up)}, ::Type{<:Duplicated{RT}}, tape, prob, sensealg::Union{Const{Nothing}, Const{<:DiffEqBase.AbstractSensitivityAlgorithm}}, u0, p, args...; kwargs...) where RT
3535
dres, clos = tape
3636
dres = dres::RT
3737
dargs = clos(dres)

0 commit comments

Comments
 (0)