-
Notifications
You must be signed in to change notification settings - Fork 25.6k
Also test predispatch serialization #121652
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
Also test predispatch serialization #121652
Conversation
[ghstack-poisoned]
[ghstack-poisoned]
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/121652
Note: Links to docs will display an error until the docs builds have been completed. ✅ You can merge normally! (2 Unrelated Failures)As of commit faad2df with merge base 39ed038 ( FLAKY - The following jobs failed but were likely due to flakiness present on trunk:
This comment was automatically generated by Dr. CI and updates every 15 minutes. |
[ghstack-poisoned]
if use_pre_dispatch: | ||
_check_graph(pre_dispatch=True) | ||
_check_graph(pre_dispatch=False) | ||
else: | ||
_check_graph(pre_dispatch=False) |
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.
if use_pre_dispatch: | |
_check_graph(pre_dispatch=True) | |
_check_graph(pre_dispatch=False) | |
else: | |
_check_graph(pre_dispatch=False) | |
if use_pre_dispatch: | |
_check_graph(pre_dispatch=True) | |
_check_graph(pre_dispatch=False) |
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.
We actually want to test both of them in use_pre_dispatch case. Meaning, we will always test aot-dispatch and pre-dispatch is just additional check
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.
yeah -- the _check_graph(pre_dispatch=False)
is outside of the if statement :P
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.
Oh hahaha
@pytorchbot merge |
Merge startedYour change will be merged once all checks pass (ETA 0-4 Hours). Learn more about merging in the wiki. Questions? Feedback? Please reach out to the PyTorch DevX Team |
In this PR, we create another dynamic test class for TestExport tests that basically serializes/deserializas pre-dispatch IR. I encountered 4 additional failures. But 3 of them are due to different operator showing up in the graph and only one legit failure which is tracked by another task internally. Pull Request resolved: #121678 Approved by: https://github.com/angelayi ghstack dependencies: #121652
Pull Request resolved: #121687 Approved by: https://github.com/avikchaudhuri ghstack dependencies: #121652, #121678
) Pull Request resolved: #121710 Approved by: https://github.com/BoyuanFeng ghstack dependencies: #121652, #121678, #121687
Stack from ghstack (oldest at bottom):