Skip to content
This repository has been archived by the owner on Dec 16, 2022. It is now read-only.

Make dist_reduce work for tensors #5147

Merged
merged 3 commits into from
Apr 23, 2021
Merged

Make dist_reduce work for tensors #5147

merged 3 commits into from
Apr 23, 2021

Conversation

AkshitaB
Copy link
Contributor

Changes proposed in this pull request:

  • Make dist_reduce work for tensors.
  • Pass the kwargs from dist_reduce_sum to dist_reduce

@AkshitaB AkshitaB changed the title Make dist_reduce_sum work for tensors Make dist_reduce work for tensors Apr 23, 2021
@AkshitaB AkshitaB requested a review from epwalsh April 23, 2021 16:21
epwalsh
epwalsh approved these changes Apr 23, 2021
View reviewed changes
Copy link
Member

@epwalsh epwalsh left a comment

Choose a reason for hiding this comment

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

LGTM!

@AkshitaB AkshitaB merged commit d11359e into main Apr 23, 2021
@AkshitaB AkshitaB deleted the update-dist-reduce branch April 23, 2021 23:22
dirkgr pushed a commit that referenced this pull request May 10, 2021
@AkshitaB @dirkgr
make dist_reduce_sum work for tensors (#5147)
7c06b49
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Reviewers

@epwalsh epwalsh epwalsh approved these changes

Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
2 participants
@AkshitaB @epwalsh