[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.

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]

We need to be merging these updates to main (we are pulling 2.6 on main branch now). Can you please rebase

Done

[svekars]

A couple of editorial suggestions.

[svekars]

Add a Prerequisites section, it could look like:

Suggested change

	#
	#
	# Prerequisites
	# ~~~~~~~~~~~~~~~
	#
	# Before starting this tutorial, make sure you have completed the following prerequisites:
	#
	# * Completed the `ONNX Script tutorial <https://github.com/microsoft/onnxscript/blob/main/docs/tutorial/index.md>`_
	# * You have the target PyTorch operator
	# * You are familiar with the implementation of the operator using `ONNX Script <https://github.com/microsoft/onnxscript>`__
	#

[titaiwangms]

Done

[svekars]

Can you also add a What you will learn section. Example:

In this tutorial, you will learn:

• How to override or add support for PyTorch operators in ONNX.
• How to integrate custom ONNX operators for specialized runtimes.
• How to implement and translate custom PyTorch operators to ONNX.

[titaiwangms]

Done

[svekars]

remove dot for consistency.

[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