KEMBAR78
Use torch with statement in torch distributed module by guangyey · Pull Request #144951 · pytorch/pytorch · GitHub
Skip to content

Conversation

@guangyey
Copy link
Collaborator

@guangyey guangyey commented Jan 16, 2025

Stack from ghstack (oldest at bottom):

Motivation

In #137678, we help use the device-agnostic APIs to generalize distributed module. As this comment said, we will use the with statement of torch.Stream once #140138 is landed.

cc @H-Huang @awgu @kwen2501 @wanchaol @fegin @fduwjj @wz337 @wconstab @d4l3k @c-p-i-o

@pytorch-bot
Copy link

pytorch-bot bot commented Jan 16, 2025

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/144951

Note: Links to docs will display an error until the docs builds have been completed.

✅ You can merge normally! (1 Unrelated Failure)

As of commit 17eb712 with merge base 0b17c09 (image):

BROKEN TRUNK - The following job failed but were present on the merge base:

👉 Rebase onto the `viable/strict` branch to avoid these failures

This comment was automatically generated by Dr. CI and updates every 15 minutes.

@pytorch-bot pytorch-bot bot added oncall: distributed Add this issue/PR to distributed oncall triage queue release notes: distributed (ddp) release notes category labels Jan 16, 2025
@guangyey guangyey added ciflow/trunk Trigger trunk jobs on your pull request topic: not user facing topic category labels Jan 16, 2025
@guangyey guangyey requested review from albanD and kwen2501 January 16, 2025 07:29
guangyey added a commit that referenced this pull request Jan 16, 2025
ghstack-source-id: a22c319
Pull Request resolved: #144951
Copy link
Contributor

@kwen2501 kwen2501 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

The change at distributed side looks good to me! I will leave it to @albanD to review the change to the Stream API.

[ghstack-poisoned]
[ghstack-poisoned]
@guangyey
Copy link
Collaborator Author

Thanks @kwen2501

Copy link
Collaborator

@albanD albanD left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

very nice cleanup!

@guangyey
Copy link
Collaborator Author

@pytorchbot merge

@pytorchmergebot
Copy link
Collaborator

Merge started

Your 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

Advanced Debugging
Check the merge workflow status
here

@github-actions github-actions bot deleted the gh/guangyey/119/head branch February 16, 2025 02:10
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ciflow/trunk Trigger trunk jobs on your pull request Merged oncall: distributed Add this issue/PR to distributed oncall triage queue open source release notes: distributed (ddp) release notes category topic: not user facing topic category

Projects

None yet

Development

Successfully merging this pull request may close these issues.

5 participants