fix

EnzymeAD · Dec 2, 2024 · 5a15880 · 5a15880
1 parent 66f5ba7
commit 5a15880
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/sugar.jl b/src/sugar.jl
@@ -288,7 +288,7 @@ grad = gradient(ReverseWithPrimal, mul, [2.0], Const([3.0]))
             push!(states, Compiler.AnyState)
         else
             state = Compiler.active_reg_inner(genty, (), nothing)
-            push!(states, Compiler.AnyState)
+            push!(states, state)
         end
     end