This repository was archived by the owner on Aug 5, 2025. It is now read-only.
update import for _export_to_torch_ir #924
Closed
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Contributor
Author
|
the test failure seems to be the inverse of the original problem - the PyTorch build running CI does not have the refactor. On my machine with latest nightly it passes: |
kwen2501
added a commit
that referenced
this pull request
Jan 2, 2024
## Description This is a copy of #924 with an addition to support both new and old import paths. --------- Co-authored-by: lessw2020 <[email protected]>
Contributor
Author
|
same PR with but updated for backwards support with this PR: #926 |
Sign up for free
to subscribe to this conversation on GitHub.
Already have an account?
Sign in.
2 participants
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
PyTorch main had a refactor for torch export related code, which breaks tracing in IR.py with latest nightlies.
This PR simply updates the import for the _export_to_torch_ir function, in order to enable access again for tracing.
Link to refactor PR for reference:
https://github.com/pytorch/pytorch/pull/114768/files
Tested on bert and bert_cpu_init as quick test.