tc

Author: clee2000

intermediate_source/memory_format_tutorial.py

@@ -341,18 +341,17 @@ def check_cl(*args, **kwargs):
     return check_cl
 
 
-old_attrs = dict()
+old_attrs = []
 
 
 def attribute(m):
-    old_attrs[m] = dict()
     for i in dir(m):
         e = getattr(m, i)
         exclude_functions = ["is_cuda", "has_names", "numel", "stride", "Tensor", "is_contiguous", "__class__"]
         if i not in exclude_functions and not i.startswith("_") and "__call__" in dir(e):
             try:
-                old_attrs[m][i] = e
                 setattr(m, i, check_wrapper(e))
+                old_attrs.append((m, i, e))
             except Exception as e:
                 print(i)
                 print(e)
@@ -372,9 +371,8 @@ def attribute(m):
 ######################################################################
 # Code below is to recover the attributes of torch.
 
-for (m, attrs) in old_attrs.items():
-    for (k, v) in attrs.items():
-        setattr(m, k, v)
+for m, i, e in reversed(old_attrs):
+    setattr(m, i, e)
 
 ######################################################################
 # Work to do