Fix the bug that __deepcopy__ does not register new node.

chinganc · chinganc · commit ab49d1a05384 · 2025-01-19T13:42:23.000-08:00
diff --git a/opto/trace/nodes.py b/opto/trace/nodes.py
@@ -466,6 +466,7 @@ def __deepcopy__(self, memo):
                 setattr(result, k, defaultdict(list))
             else:
                 setattr(result, k, copy.deepcopy(v, memo))
+        GRAPH.register(result)
         return result
 
     def lt(self, other):
