Skip to content

support dist.reduce_scatter_tensor#7950

Merged
zpcore merged 1 commit intomasterfrom
piz/cop
Sep 4, 2024
Merged

support dist.reduce_scatter_tensor#7950
zpcore merged 1 commit intomasterfrom
piz/cop

Conversation

@zpcore
Copy link
Copy Markdown
Member

@zpcore zpcore commented Sep 3, 2024

Support of torch.distributed.reduce_scatter_tensor() for both dynamo and nondynamo case.

@zpcore zpcore requested a review from will-cromar September 3, 2024 23:16
@zpcore zpcore marked this pull request as ready for review September 3, 2024 23:17
@zpcore zpcore changed the title support dist.reduce_scatter support dist.reduce_scatter_tensor Sep 3, 2024
Copy link
Copy Markdown
Collaborator

@will-cromar will-cromar left a comment

Choose a reason for hiding this comment

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

Thanks!

@zpcore zpcore merged commit b1fbb74 into master Sep 4, 2024
@zpcore zpcore deleted the piz/cop branch September 4, 2024 17:20
@miladm miladm added the usability Bugs/features related to improving the usability of PyTorch/XLA label Nov 22, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

usability Bugs/features related to improving the usability of PyTorch/XLA

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants