xla
3cd39ea5
- Add separate BuildAllGatherCoalesced builder and AllGatherCoalesced class
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
1 year ago
Add separate BuildAllGatherCoalesced builder and AllGatherCoalesced class
References
#5950 - Add all-gather coalescing for FSDP/ZeRO1
Author
jeffhataws
Parents
8e6715e7
Loading