Skip to content

Navigation Menu

Explore
By company size
By use case
By industry
View all solutions
Topics
- AI
- DevOps
- Security
- Software Development
- View all
Explore
- GitHub Sponsors
  Fund open source developers
- The ReadME Project
  GitHub community articles
Repositories
- Enterprise platform
  AI-powered developer platform
Available add-ons
Pricing

Search code, repositories, users, issues, pull requests...

Search

Clear

Search syntax tips

Provide feedback

We read every piece of feedback, and take your input very seriously.

Include my email address so I can be contacted

Saved searches

Use saved searches to filter your results more quickly

Name

Query

To see all available qualifiers, see our documentation.

You signed in with another tab or window. Reload to refresh your session. You signed out in another tab or window. Reload to refresh your session. You switched accounts on another tab or window. Reload to refresh your session.

Dismiss alert

pytorch / xla Public

Notifications You must be signed in to change notification settings
Fork 486
Star 2.5k

Code
Issues 592
Pull requests 148
Actions
Projects 11
Security
Insights

Additional navigation options

Code
Issues
Pull requests
Actions
Projects
Security
Insights

Add reduce-scatter coalescing for FSDP/ZeRO1#5956

Merged

JackCaoG merged 4 commits intomasterpytorch/xla:masterfrom cc_coalesce_reducescatterpytorch/xla:cc_coalesce_reducescatterCopy head branch name to clipboard

Dec 7, 2023

+366-19

Conversation Commits4 (4)Checks Files changed

Commits

Commits on Dec 5, 2023

Add reduce-scatter coalescence support for FSDP.
jeffhataws
committed
Fix error checking when input is list and output is specified
jeffhataws
committed

Commits on Dec 6, 2023

Add reduce-scatter test with list of tensors as input/output
jeffhataws
committed
For now, error when reduce-scatter coaelesce xm API is used with output!=None

jeffhataws
authored and
Arjunbala
committed

Footer

© 2024 GitHub, Inc.

Footer navigation

Terms
Privacy
Security
Status
Docs
Contact

You can’t perform that action at this time.