multiply by dret

SciML · Sep 27, 2024 · 9382e4b · 9382e4b
1 parent 181748f
commit 9382e4b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ext/DiffEqBaseEnzymeExt.jl b/ext/DiffEqBaseEnzymeExt.jl
@@ -90,8 +90,8 @@ function EnzymeRules.reverse(config::Enzyme.EnzymeRules.ConfigWidth{1},
         func::Const{typeof(DiffEqBase.fastpow)}, dret::Active, tape, _x::Active, _y::Active)
     x = _x.val
     y = _y.val
-    dxval =  y * (fastpow(x,y - 1))
-    dyval = x isa Real && x<=0 ? Base.oftype(float(x), NaN) :  (fastpow(x,y))*log(x)
+    dxval =  dret.val * y * (fastpow(x,y - 1))
+    dyval = x isa Real && x<=0 ? Base.oftype(float(x), NaN) : dret.val * (fastpow(x,y))*log(x)
     return (dxval, dyval)
 end