Skip to content

[Bug fix] Create gloo subgroups among their members only - #369

Merged
stephen-dwq merged 1 commit into
mainfrom
fix/gloo-subgroup-hang
Oct 8, 2026
Merged

stephen-dwq merged 1 commit into
mainfrom
fix/gloo-subgroup-hang

Conversation

@NSagan271

Copy link
Copy Markdown
Collaborator

What does this PR do?

Fixes a bug introduced in #258 that prevented Qwen3-Omni with Thinker TP2 from starting.

With the default process group bound to a device, creating a subgroup on a rank outside it runs an ncclCommSplit that every world rank must join. For a gloo subgroup the members never call it, so a deployment with a rank outside every parallel group (Qwen3-Omni's Thinker on TP2 with the Talker on its own GPU) hung at startup: the non-member blocked in new_group while the members waited at the startup barrier. use_local_synchronization makes the gloo groups members-only and names them by their ranks, so the shared group counter stays in step.

How was it tested?

Verified Qwen3-Omni with Thinker TP 2 starts and runs as normal.

Checklist

  • ruff check . passes
  • Added or updated tests / docs where relevant

With the default process group bound to a device, creating a subgroup on a
rank outside it runs an ncclCommSplit that every world rank must join. For
a gloo subgroup the members never call it, so a deployment with a rank
outside every parallel group (Qwen3-Omni's Thinker on TP2 with the Talker
on its own GPU) hung at startup: the non-member blocked in new_group while
the members waited at the startup barrier. use_local_synchronization makes
the gloo groups members-only and names them by their ranks, so the shared
group counter stays in step.

Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
@NSagan271
NSagan271 marked this pull request as ready for review October 8, 2026 19:15

@stephen-dwq stephen-dwq left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

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

lgtm

@stephen-dwq
stephen-dwq merged commit f49a1e5 into main Oct 8, 2026
4 checks passed
@stephen-dwq
stephen-dwq deleted the fix/gloo-subgroup-hang branch October 8, 2026 19:22
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