-
Notifications
You must be signed in to change notification settings - Fork 22.6k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[ONNX] Move symbolic export to separate file #95650
Conversation
[ghstack-poisoned]
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/95650
Note: Links to docs will display an error until the docs builds have been completed. ✅ No FailuresAs of commit 54e5377: This comment was automatically generated by Dr. CI and updates every 15 minutes. |
[ghstack-poisoned]
Move things around in the effort of preparing to refactor the code structure. [ghstack-poisoned]
|
||
|
||
@_beartype.beartype | ||
def _trace_into_fx_graph_via_fx_symbolic_trace( |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Make it a FX frontend?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Will attempt in a follow up.
|
||
|
||
@_beartype.beartype | ||
def _move_placeholder_to_front(graph_module: torch.fx.GraphModule) -> None: |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
There are some graph transforms in this file. They are good candidates for new Transform
classes.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Done in the pass PR.
Move things around in the effort of preparing to refactor the code structure. [ghstack-poisoned]
Move things around in the effort of preparing to refactor the code structure. Pull Request resolved: pytorch/pytorch#95650 Approved by: https://github.com/titaiwangms
Move things around in the effort of preparing to refactor the code structure. Pull Request resolved: pytorch/pytorch#95650 Approved by: https://github.com/titaiwangms
Move things around in the effort of preparing to refactor the code structure. Pull Request resolved: pytorch#95650 Approved by: https://github.com/titaiwangms
Stack from ghstack (oldest at bottom):
torch.onnx._internal.fx
if ImportError #95686Move things around in the effort of preparing to refactor
the code structure.