From ef1b2da2eba1920e4371037f34fe94dcffc80def Mon Sep 17 00:00:00 2001
From: Brian Chen <ToucheSir@users.noreply.github.com>
Date: Mon, 8 Jan 2024 12:09:21 -0800
Subject: [PATCH] Missed grad_mut

---
 src/lib/base.jl | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/lib/base.jl b/src/lib/base.jl
index 86aeb6095..4ee5947a5 100644
--- 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(_)