Missed grad_mut

FluxML · Jan 8, 2024 · ef1b2da · ef1b2da
1 parent 49592f7
commit ef1b2da
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/lib/base.jl b/src/lib/base.jl
@@ -18,6 +18,9 @@ end
 end
 
 # IdSet (needed for nested AD with implicit params)
+
+grad_mut(::IdSet) = IdSet()
+
 function _pullback(cx::AContext, ::typeof(push!), s::IdSet, @nospecialize(x))
   res = push!(s, x)
   function idset_push!_pullback(_)