Skip to content
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 our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Collective's PREMUL_SUM support with PyTorch 1.13 #15201

Merged
merged 7 commits into from Oct 20, 2022
Merged

Conversation

carmocca
Copy link
Member

@carmocca carmocca commented Oct 20, 2022

What does this PR do?

Re-do of #15167
Workaround for pytorch/pytorch#87191
PREMUL_SUM was added in pytorch/pytorch#84243
Things might change after pytorch/pytorch#87303, it's unclear whether the API will change before 1.13 is released.

Does your PR introduce any breaking changes? If yes, please list them.

None

cc @Borda @justusschock @awaelchli @rohitgr7 @carmocca

@carmocca carmocca added refactor fabric lightning.fabric.Fabric labels Oct 20, 2022
@carmocca carmocca self-assigned this Oct 20, 2022
@codecov
Copy link

codecov bot commented Oct 20, 2022

Codecov Report

Merging #15201 (f247462) into master (0bafb38) will increase coverage by 21%.
The diff coverage is 92%.

Additional details and impacted files
@@            Coverage Diff            @@
##           master   #15201     +/-   ##
=========================================
+ Coverage      62%      84%    +21%     
=========================================
  Files         413      288    -125     
  Lines       30309    21993   -8316     
=========================================
- Hits        18817    18378    -439     
+ Misses      11492     3615   -7877     

@Borda Borda requested review from Borda and otaj October 20, 2022 07:59
@Borda Borda enabled auto-merge (squash) October 20, 2022 08:00
@mergify mergify bot added ready PRs ready to be merged has conflicts and removed ready PRs ready to be merged labels Oct 20, 2022
@mergify mergify bot added ready PRs ready to be merged has conflicts and removed has conflicts ready PRs ready to be merged labels Oct 20, 2022
@mergify mergify bot added ready PRs ready to be merged and removed has conflicts labels Oct 20, 2022
@mergify mergify bot added ready PRs ready to be merged and removed ready PRs ready to be merged labels Oct 20, 2022
@carmocca carmocca added this to the v1.8 milestone Oct 20, 2022
@Borda Borda merged commit b866dc3 into master Oct 20, 2022
@Borda Borda deleted the feat/premu-sum branch October 20, 2022 12:36
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
fabric lightning.fabric.Fabric ready PRs ready to be merged refactor
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

4 participants