Skip to content

Propagates requires_grad over to AllReduce output#6326

Merged
yeounoh merged 1 commit intomasterfrom
all_reduce_output_grad
Jan 19, 2024
Merged

Propagates requires_grad over to AllReduce output#6326
yeounoh merged 1 commit intomasterfrom
all_reduce_output_grad

Conversation

@yeounoh
Copy link
Copy Markdown
Contributor

@yeounoh yeounoh commented Jan 18, 2024

Addresses #6319 cc @aws-rhsoln

@yeounoh yeounoh self-assigned this Jan 18, 2024
Copy link
Copy Markdown
Collaborator

@alanwaketan alanwaketan left a comment

Choose a reason for hiding this comment

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

LGTM.

@yeounoh yeounoh merged commit 40727e4 into master Jan 19, 2024
jeffhataws pushed a commit to jeffhataws/xla that referenced this pull request Jan 19, 2024
jeffhataws added a commit to jeffhataws/xla that referenced this pull request Jan 22, 2024
Propagates requires_grad over to AllReduce output (pytorch#6326)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants