Skip to content

Fix gloo backend error in test_load_checkpoint_and_dispatch_with_broa… - #4056

Merged
SunMarc merged 1 commit into
huggingface:mainfrom
kaixuanliu:xpu-hybrid
Jun 1, 2026
Merged

SunMarc merged 1 commit into
huggingface:mainfrom
kaixuanliu:xpu-hybrid

Conversation

@kaixuanliu

Copy link
Copy Markdown
Contributor

In test case: tests/test_load_checkpoint_and_dispatch_with_broadcast.py::TestLoadCheckpointAndDispatchWithBroadcast::test_load_checkpoint_and_dispatch_ddp
the DDP test only registered xccl(using default backend), but DDP's logger requires a gloo backend on the PG (which xccl doesn't fall back to, unlike NCCL), so init now uses the hybrid cpu:gloo,xpu:xccl — matching the FSDP setup already in state.py: L243

…dcast on XPU

Signed-off-by: kaixuanliu <kaixuan.liu@intel.com>
@kaixuanliu

Copy link
Copy Markdown
Contributor Author

@SunMarc pls help review, thx!

@SunMarc SunMarc 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.

Thanks !

@HuggingFaceDocBuilderDev

Copy link
Copy Markdown

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

@kaixuanliu

kaixuanliu commented Jun 1, 2026 •

Copy link
Copy Markdown
Contributor Author

The failed CI should not be introduced by this PR. Seems dependant transformers is not correctly installed.

@SunMarc
SunMarc merged commit 4ad096a into huggingface:main Jun 1, 2026
16 of 25 checks passed
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.

3 participants