[test] memory format tutorial

.ci/docker/requirements.txt

@@ -2,7 +2,7 @@
 # Refer to ./jenkins/build.sh for tutorial build instructions
 
 sphinx==5.0.0
-sphinx-gallery==0.11.1
+sphinx-gallery==0.19.0
 sphinx_design
 docutils==0.16
 sphinx-copybutton

.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]:

conf.py

@@ -126,6 +126,7 @@ def reset_seeds(gallery_conf, fname):
     'pypandoc': {'extra_args': ['--mathjax', '--toc'],
                  'filters': ['.jenkins/custom_pandoc_filter.py'],
     },
+    'parallel': True,
 }
 
 html_baseurl = 'https://pytorch.org/tutorials/' # needed for sphinx-sitemap

custom_directives.py

@@ -88,7 +88,7 @@ def run(self):
             if 'intro' in self.options:
                 intro = self.options['intro'][:195] + '...'
             else:
-                _, blocks = sphinx_gallery.gen_rst.split_code_and_text_blocks(abs_fname)
+                _, blocks = sphinx_gallery.py_source_parser.split_code_and_text_blocks(abs_fname)
                 intro, _ = sphinx_gallery.gen_rst.extract_intro_and_title(abs_fname, blocks[0][1])
 
             thumbnail_rst = ''

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