Justin Chu
3771380f83
[ONNX] Hide draft export under a flag ( #162225 )
...
Use `TORCH_ONNX_ENABLE_DRAFT_EXPORT` to control whether draft_export should be used as a strategy in onnx export.
Follow up of https://github.com/pytorch/pytorch/pull/161454
Pull Request resolved: https://github.com/pytorch/pytorch/pull/162225
Approved by: https://github.com/xadupre , https://github.com/titaiwangms
2025-09-05 19:54:50 +00:00
Justin Chu
2dccd70ef0
[ONNX] Clean up legacy dynamo export code ( #149745 )
...
Clean up code that is unused and obsolete. The public `torch.onnx.dynamo_export` is kept for now but the legacy implementation is removed.
Remove public option classes and OnnxRegistry that have been deprecated.
Users: use torch.onnx.export(…, dynamo=True).
Pull Request resolved: https://github.com/pytorch/pytorch/pull/149745
Approved by: https://github.com/titaiwangms , https://github.com/cyyever
2025-03-23 19:35:16 +00:00
Justin Chu
af43b445a5
[ONNX] Set USE_EXPERIMENTAL_LOGIC to True ( #137296 )
...
This sets dynamo_export to use the new export logic. The legacy dynamo export logic will be removed as a follow up.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/137296
Approved by: https://github.com/titaiwangms
2025-01-28 22:35:11 +00:00
Justin Chu
0c491702c4
[ONNX] Define the TORCH_ONNX_USE_EXPERIMENTAL_LOGIC flag ( #132299 )
...
Define the `TORCH_ONNX_USE_EXPERIMENTAL_LOGIC` flag to allow for enabling the new torch.onnx logic and hiding them during migration and testing. The actual logic migration will happen after.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/132299
Approved by: https://github.com/titaiwangms
2024-08-02 17:06:11 +00:00
