Merge branch 'main' into csl/fix_isolation_macs

Author: clee2000

.ci/docker/requirements.txt

@@ -2,7 +2,7 @@
 # Refer to ./jenkins/build.sh for tutorial build instructions
 
 sphinx==5.3.0
-sphinx-gallery==0.11.1
+sphinx-gallery==0.17.1
 sphinx-reredirects==0.1.4
 sphinx-design==0.4.0
 docutils==0.16
@@ -31,7 +31,8 @@ pytorch-lightning
 torchx
 torchrl==0.7.2
 tensordict==0.7.2
-ax-platform>=0.4.0
+# For ax_multiobjective_nas_tutorial.py
+ax-platform>=0.4.0,<0.5.0
 nbformat>=5.9.2
 datasets
 transformers

Makefile

@@ -61,23 +61,6 @@ download:
 	wget -nv -N https://s3.amazonaws.com/pytorch-tutorial-assets/cornell_movie_dialogs_corpus_v2.zip -P $(DATADIR)
 	unzip $(ZIPOPTS) $(DATADIR)/cornell_movie_dialogs_corpus_v2.zip -d beginner_source/data/
 
-	# Download model for advanced_source/dynamic_quantization_tutorial.py
-	wget -nv -N https://s3.amazonaws.com/pytorch-tutorial-assets/word_language_model_quantize.pth -P $(DATADIR)
-	cp $(DATADIR)/word_language_model_quantize.pth advanced_source/data/word_language_model_quantize.pth
-
-	# Download data for advanced_source/dynamic_quantization_tutorial.py
-	wget -nv -N https://s3.amazonaws.com/pytorch-tutorial-assets/wikitext-2.zip -P $(DATADIR)
-	unzip $(ZIPOPTS) $(DATADIR)/wikitext-2.zip -d advanced_source/data/
-
-	# Download model for advanced_source/static_quantization_tutorial.py
-	wget -nv -N https://download.pytorch.org/models/mobilenet_v2-b0353104.pth -P $(DATADIR)
-	cp $(DATADIR)/mobilenet_v2-b0353104.pth advanced_source/data/mobilenet_pretrained_float.pth
-
-
-	# Download model for prototype_source/graph_mode_static_quantization_tutorial.py
-	wget -nv -N https://download.pytorch.org/models/resnet18-5c106cde.pth -P $(DATADIR)
-	cp $(DATADIR)/resnet18-5c106cde.pth prototype_source/data/resnet18_pretrained_float.pth
-
 	# Download PennFudanPed dataset for intermediate_source/torchvision_tutorial.py
 	wget https://www.cis.upenn.edu/~jshi/ped_html/PennFudanPed.zip -P $(DATADIR)
 	unzip -o $(DATADIR)/PennFudanPed.zip -d intermediate_source/data/

_static/img/thumbnails/cropped/graph-mode-dynamic-bert.png

@@ -88,7 +88,8 @@ 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)
+                block_parser = sphinx_gallery.gen_rst.BlockParser(abs_fname, {"filetype_parsers": {}})
+                _, blocks, _ = block_parser.split_code_and_text_blocks(abs_fname)
                 intro, _ = sphinx_gallery.gen_rst.extract_intro_and_title(abs_fname, blocks[0][1])
 
             thumbnail_rst = ''

custom_directives.py

@@ -4,42 +4,18 @@ Prototype Tutorials
            Profiling PyTorch RPC-Based Workloads
            https://github.com/pytorch/tutorials/blob/main/prototype_source/distributed_rpc_profiling.rst
 
-2. graph_mode_static_quantization_tutorial.py
-	   Graph Mode Post Training Static Quantization in PyTorch
-	   https://pytorch.org/tutorials/prototype/graph_mode_static_quantization_tutorial.html
-
-3. graph_mode_dynamic_bert_tutorial.rst
-	   Graph Mode Dynamic Quantization on BERT
-	   https://github.com/pytorch/tutorials/blob/main/prototype_source/graph_mode_dynamic_bert_tutorial.rst
-
-4. numeric_suite_tutorial.py
-	   PyTorch Numeric Suite Tutorial
-	   https://github.com/pytorch/tutorials/blob/main/prototype_source/numeric_suite_tutorial.py
-
-5. torchscript_freezing.py
+2. torchscript_freezing.py
 	   Model Freezing in TorchScript
 	   https://github.com/pytorch/tutorials/blob/main/prototype_source/torchscript_freezing.py
 
-6. vulkan_workflow.rst
+3. vulkan_workflow.rst
            Vulkan Backend User Workflow
-           https://pytorch.org/tutorials/intermediate/vulkan_workflow.html
-
-7. fx_graph_mode_ptq_static.rst
-	   FX Graph Mode Post Training Static Quantization
-	   https://pytorch.org/tutorials/prototype/fx_graph_mode_ptq_static.html
-
-8. fx_graph_mode_ptq_dynamic.py
-	   FX Graph Mode Post Training Dynamic Quantization
-	   https://pytorch.org/tutorials/prototype/fx_graph_mode_ptq_dynamic.html
-
-9. fx_graph_mode_quant_guide.py
-	   FX Graph Mode Quantization User Guide
-	   https://pytorch.org/tutorials/prototype/fx_graph_mode_quant_guide.html
-
-10 flight_recorder_tutorial.rst
+           https://pytorch.org/tutorials/prototype/vulkan_workflow.html
+           
+4. flight_recorder_tutorial.rst
 	   Flight Recorder User Guide
 	   https://pytorch.org/tutorials/prototype/flight_recorder_tutorial.html
 
-11 python_extension_autoload.rst
+5. python_extension_autoload.rst
 	   Autoloading Out-of-Tree Extension
 	   https://pytorch.org/tutorials/prototype/python_extension_autoload.html