Skip to content

[CUDA][NCCL] group split - #3172

Merged
nastya236 merged 4 commits into
ml-explore:mainfrom
nastya236:nccLgroup_split
Feb 26, 2026
Merged

[CUDA][NCCL] group split #3172
nastya236 merged 4 commits into
ml-explore:mainfrom
nastya236:nccLgroup_split

Conversation

@nastya236

Copy link
Copy Markdown
Collaborator

Add group split to NCCL backend for multi-parallelism support.

Refactored communicator creation by introducing a dedicated helper struct. It is probably not very important but I also updated the communicator storage from a raw pointer to a shared one for better ownership management.

@nastya236
nastya236 marked this pull request as ready for review February 25, 2026 17:19
@nastya236
nastya236 marked this pull request as draft February 25, 2026 17:28
@nastya236
nastya236 marked this pull request as ready for review February 25, 2026 21:59

@angeloskath angeloskath left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

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

Looks great!

@nastya236
nastya236 merged commit 4e00919 into ml-explore:main Feb 26, 2026
16 checks passed
jzdziarski pushed a commit to jzdziarski/mlx that referenced this pull request Jul 26, 2026
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.

2 participants