diff --git a/.jenkins/validate_tutorials_built.py b/.jenkins/validate_tutorials_built.py index f5cd187dbc..5c9e60e90b 100644 --- a/.jenkins/validate_tutorials_built.py +++ b/.jenkins/validate_tutorials_built.py @@ -53,7 +53,6 @@ "intermediate_source/tensorboard_profiler_tutorial", # reenable after 2.0 release. "advanced_source/semi_structured_sparse", # reenable after 3303 is fixed. "intermediate_source/torchrec_intro_tutorial", # reenable after 3302 is fixe - "intermediate_source/memory_format_tutorial", # causes other tutorials like torch_logs fail. "state" issue, reseting dynamo didn't help ] def tutorial_source_dirs() -> List[Path]: diff --git a/intermediate_source/memory_format_tutorial.py b/intermediate_source/memory_format_tutorial.py index 26bc5c9d53..b73f3c674e 100644 --- a/intermediate_source/memory_format_tutorial.py +++ b/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