From 549bdfcb6544938f219e178d545be219b765728b Mon Sep 17 00:00:00 2001 From: Matt Bossart Date: Sun, 9 Jun 2024 11:31:58 -0400 Subject: [PATCH] fix make_zero! --- ext/DiffEqBaseEnzymeExt.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ext/DiffEqBaseEnzymeExt.jl b/ext/DiffEqBaseEnzymeExt.jl index b31c6f26c..2b2b7c001 100644 --- a/ext/DiffEqBaseEnzymeExt.jl +++ b/ext/DiffEqBaseEnzymeExt.jl @@ -49,7 +49,7 @@ function Enzyme.EnzymeRules.reverse(config::Enzyme.EnzymeRules.ConfigWidth{1}, end ptr.dval .+= darg end - Enzyme.make_zero!(dres) + Enzyme.make_zero!(dres.u) return ntuple(_ -> nothing, Val(length(args) + 4)) end