- Notifications
You must be signed in to change notification settings - Fork26.3k
[c10d] init_process_group supports index-only device id#156214
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 ourterms of service andprivacy statement. We’ll occasionally send you account related emails.
Already on GitHub?Sign in to your account
Uh oh!
There was an error while loading.Please reload this page.
Conversation
pytorch-botbot commentedJun 17, 2025 • edited
Loading Uh oh!
There was an error while loading.Please reload this page.
edited
Uh oh!
There was an error while loading.Please reload this page.
🔗 Helpful Links🧪 See artifacts and rendered test results athud.pytorch.org/pr/156214
Note: Links to docs will display an error until the docs builds have been completed. ⏳ No Failures, 33 PendingAs of commitc0ac973 with merge basefbbab79 ( This comment was automatically generated by Dr. CI and updates every 15 minutes. |
kwen2501 commentedJun 17, 2025
albanD 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.
nice!
| group_name:str="", | ||
| pg_options:Optional[Any]=None, | ||
| device_id:Optional[torch.device]=None, | ||
| device_id:Optional[Union[torch.device,int]]=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.
nit: you can usetorch.types.Device
kwen2501 commentedJun 20, 2025
@pytorchbot merge -f "Minor lint; all tests passed" |
pytorchmergebot commentedJun 20, 2025
Merge startedYour change will be merged immediately since you used the force (-f) flag,bypassing any CI checks (ETA: 1-5 minutes). Please use Learn more about merging in thewiki. Questions? Feedback? Please reach out to thePyTorch DevX Team |
pytorchmergebot commentedJun 20, 2025
Merge failedReason: Command Details for Dev Infra teamRaised byworkflow job |
kwen2501 commentedJun 21, 2025
@pytorchbot merge -f "Minor rebase; all tests passed" |
pytorchmergebot commentedJun 21, 2025
Merge startedYour change will be merged immediately since you used the force (-f) flag,bypassing any CI checks (ETA: 1-5 minutes). Please use Learn more about merging in thewiki. Questions? Feedback? Please reach out to thePyTorch DevX Team |
Uh oh!
There was an error while loading.Please reload this page.
Stack fromghstack (oldest at bottom):
Before:
After:
That is,
init_process_groupcheckstorch.accelerator.current_accelerator()internally.cc@H-Huang@awgu@wanchaol@fegin@fduwjj@wz337@wconstab@d4l3k