Fix export tutorial

Author: angelayi

.jenkins/validate_tutorials_built.py

@@ -23,7 +23,6 @@
     "intermediate_source/dqn_with_rnn_tutorial", #not working on 2.8 release reenable after 3514
     "intermediate_source/mnist_train_nas",  # used by ax_multiobjective_nas_tutorial.py
     "intermediate_source/torch_compile_conv_bn_fuser",
-    "intermediate_source/_torch_export_nightly_tutorial",  # does not work on release
     "advanced_source/usb_semisup_learn", # fails with CUDA OOM error, should try on a different worker
     "unstable_source/gpu_direct_storage", # requires specific filesystem + GPUDirect Storage to be set up
     "recipes_source/recipes/tensorboard_with_pytorch",

conf.py

@@ -181,7 +181,6 @@ def wrapper(*args, **kwargs):
         "# https://docs.pytorch.org/tutorials/beginner/colab\n"
         "%matplotlib inline"
     ),
-    "ignore_pattern": r"_torch_export_nightly_tutorial.py",
     "pypandoc": {
         "extra_args": ["--mathjax", "--toc"],
         "filters": [".jenkins/custom_pandoc_filter.py"],

intermediate_source/torch_export_nightly_tutorial.rst

@@ -843,10 +843,7 @@ def forward(self, x):
 print(torch.ops.aten.add_.Tensor._schema.is_mutable)
 
 ######################################################################
-# This generic IR can be used to train in eager PyTorch Autograd. This IR can be
-# more explicitly reached through the API ``torch.export.export_for_training``,
-# which was introduced in PyTorch 2.5, but calling ``torch.export.export``
-# should produce the same graph as of PyTorch 2.6.
+# This generic IR can be used to train in eager PyTorch Autograd.
 
 class DecompExample(torch.nn.Module):
     def __init__(self) -> None:
@@ -859,7 +856,7 @@ def forward(self, x):
         x = self.bn(x)
         return (x,)
 
-ep_for_training = torch.export.export_for_training(DecompExample(), (torch.randn(1, 1, 3, 3),))
+ep_for_training = torch.export.export(DecompExample(), (torch.randn(1, 1, 3, 3),))
 print(ep_for_training.graph)
 
 ######################################################################