Skip to content

Commit 8dd1012

Browse files
Update ext/DiffEqBaseEnzymeExt.jl
1 parent 84a8d6d commit 8dd1012

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ext/DiffEqBaseEnzymeExt.jl

Lines changed: 1 addition & 1 deletion
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, 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{<: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)

0 commit comments

Comments
 (0)