fixup

FluxML · Jan 8, 2024 · 49592f7 · 49592f7
1 parent 4638239
commit 49592f7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/lib/base.jl b/src/lib/base.jl
@@ -21,7 +21,7 @@ end
 function _pullback(cx::AContext, ::typeof(push!), s::IdSet, @nospecialize(x))
   res = push!(s, x)
   function idset_push!_pullback(_)
-    Δ = pop!(grad_mut(cx, d), x, nothing)
+    Δ = pop!(grad_mut(cx, s), x, nothing)
     (nothing, Δ, nothing)
   end
   return res, idset_push!_pullback