[ONNX] Update API to torch.onnx.export(..., dynamo=True)

[titaiwangms]

Fixes #3200

Description

Checklist

• The issue that is being fixed is referred in the description (see above "Fixes #ISSUE_NUMBER")
• Only one issue is addressed in this pull request
• Labels from the issue that this PR is fixing are added to this pull request
• No unnecessary issues are included into this pull request.

TODO: The registry doc also needs an update.

cc @justinchuby

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/tutorials/3223

• 📄 Preview Python docs built from this PR

Note: Links to docs will display an error until the docs builds have been completed.

✅ You can merge normally! (1 Unrelated Failure)

As of commit bfad889 with merge base 469d95b ():

BROKEN TRUNK - The following job failed but were present on the merge base:

👉 Rebase onto the `viable/strict` branch to avoid these failures

• Check spelling / pyspelling (gh) (trunk failure)
  Process completed with exit code 1.

This comment was automatically generated by Dr. CI and updates every 15 minutes.

[titaiwangms]

@svekars PTAL

[justinchuby]

I don’t see print outputs here either https://pytorch.org/tutorials/beginner/onnx/intro_onnx.html. Do we need to do something to enable it? (If there is an option)

[justinchuby]

They have prints here https://pytorch.org/tutorials/intermediate/torchrec_intro_tutorial.html

[titaiwangms]

They have prints here https://pytorch.org/tutorials/intermediate/torchrec_intro_tutorial.html

Yeah, I don't know. I tried a local build, and registry has the print out while control flow does not.

[titaiwangms]

@svekars Should we skip link checker on this PR, since it's newly added page? Or is there anything I missed to enable the link?

[justinchuby]

@svekars bump

[svekars]

skip-linkcheck because the failing link will become live after PR is merged.

[titaiwangms]

@svekars Sorry, could you approve the workflows again? I missed the misspelled words. Could you also point me to the local repro of misspelled word checks? Thank you!

[titaiwangms]

TODO: Update registry tutorial with triton support: pytorch/pytorch#147374

[titaiwangms]

@svekars Sorry, could you approve the workflows again? I missed the misspelled words. Could you also point me to the local repro of misspelled word checks? Thank you!

@svekars PTAL

[svekars]

do you want to add a customcard for this tutorial in index.rst (like this) so it's discoverable on the landing page under the ONNX selector

[svekars]

Maybe you can you submit a follow up for this.

[titaiwangms]

Nice suggestion!

[titaiwangms]

I will do it as a follow up.

[titaiwangms]

@svekars Thanks for helping me on this PR!

I set up the exact same bash script test for pyspelling locally, and it gave me the same error as the one on CI when I run it on main branch with ONNX pages. Do you recommend us just remove toctree? Is there another preferrable way to keep the functionality of showing related pages to user in the bottom of the current tutorial page? Thanks!

[svekars]

@svekars Thanks for helping me on this PR!

I set up the exact same bash script test for pyspelling locally, and it gave me the same error as the one on CI when I run it on main branch with ONNX pages. Do you recommend us just remove toctree? Is there another preferrable way to keep the functionality of showing related pages to user in the bottom of the current tutorial page? Thanks!

Can you point me to the error? Spellcheck can be ignored for now.

[titaiwangms]

It's under pyspelling CI test.

Checking spelling in beginner_source/onnx/export_control_flow_model_to_onnx_tutorial.py
Spelling errors found in beginner_source/onnx/export_control_flow_model_to_onnx_tutorial.py:
Misspelled words:
<url-free> beginner_source/onnx/export_control_flow_model_to_onnx_tutorial.py(161): html>body>p
--------------------------------------------------------------------------------
onnx
toc
txt
--------------------------------------------------------------------------------

Misspelled words:
<url-free> beginner_source/onnx/export_control_flow_model_to_onnx_tutorial.py(161): html>body>p
--------------------------------------------------------------------------------
toctree
--------------------------------------------------------------------------------

!!!Spelling check failed!!!
Checking spelling in beginner_source/onnx/export_simple_model_to_onnx_tutorial.py
Spelling errors found in beginner_source/onnx/export_simple_model_to_onnx_tutorial.py:
Misspelled words:
<url-free> beginner_source/onnx/export_simple_model_to_onnx_tutorial.py(202): html>body>p
--------------------------------------------------------------------------------
onnx
toc
txt
--------------------------------------------------------------------------------

Misspelled words:
<url-free> beginner_source/onnx/export_simple_model_to_onnx_tutorial.py(202): html>body>p
--------------------------------------------------------------------------------
toctree
--------------------------------------------------------------------------------

!!!Spelling check failed!!!
Checking spelling in beginner_source/onnx/intro_onnx.py
Spelling errors found in beginner_source/onnx/intro_onnx.py:
Misspelled words:
<url-free> beginner_source/onnx/intro_onnx.py(1): : html>body>p
--------------------------------------------------------------------------------
onnx
toc
txt
--------------------------------------------------------------------------------

Misspelled words:
<url-free> beginner_source/onnx/intro_onnx.py(1): : html>body>p
--------------------------------------------------------------------------------
toctree
--------------------------------------------------------------------------------

!!!Spelling check failed!!!
Checking spelling in beginner_source/onnx/onnx_registry_tutorial.py
Spelling errors found in beginner_source/onnx/onnx_registry_tutorial.py:
Misspelled words:
<url-free> beginner_source/onnx/onnx_registry_tutorial.py(247): html>body>p
--------------------------------------------------------------------------------
onnx
toc
txt
--------------------------------------------------------------------------------

Misspelled words:
<url-free> beginner_source/onnx/onnx_registry_tutorial.py(247): html>body>p
--------------------------------------------------------------------------------
toctree
--------------------------------------------------------------------------------

[svekars]

Can ignore spellcheck for now.