-
Notifications
You must be signed in to change notification settings - Fork 26.3k
[dtensor] enable op db tests by using multithreaded test case #92198
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
Conversation
[ghstack-poisoned]
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/92198
Note: Links to docs will display an error until the docs builds have been completed. ✅ No FailuresAs of commit a27147d: This comment was automatically generated by Dr. CI and updates every 15 minutes. |
…ase" [ghstack-poisoned]
XilunWu
left a comment
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.
good to see that op db tests are now enabled!! Is there a way to see how much time more spent on CI now?
…ase" [ghstack-poisoned]
…ase" Time comparison between using MultithreadedTestCase and MultiProcessTestCase on op db tests is amazing! using MultiThreadTestCase on a AWS dev node: ``` time pytest test/distributed/_tensor/test_dtensor_ops.py ============= 175 passed, 42 skipped, 397 xfailed in 80.30s (0:01:20) ======= real 1m22.330s user 1m38.782s sys 0m18.762s ``` MultiProcessTestCase spends from 40mins to more than 1h, even if using pytest parallel testing tools. [ghstack-poisoned]
|
@pytorchbot merge |
updated in the summary. |
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 |
|
The speedup of testing is quite impressive!! Congrats! |
Delete this test as it's now enabled in PyTorch directly after pytorch/pytorch#92198 [ghstack-poisoned]
Delete this test as it's now enabled in PyTorch directly after pytorch/pytorch#92198 ghstack-source-id: 3a58d5b Pull Request resolved: #722
Stack from ghstack (oldest at bottom):
Time comparison between using MultithreadedTestCase and MultiProcessTestCase on op db tests is amazing!
using MultiThreadTestCase on a AWS dev node:
MultiProcessTestCase spends from 40mins to more than 1h, even if using pytest parallel testing tools.