Skip to content

Commit 7925a2a

Browse files
committed
fix backend
Signed-off-by: jiqing-feng <jiqing.feng@intel.com>
1 parent 5d695a5 commit 7925a2a

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

tests/fsdp_state_dict_save.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@
2323
def _get_device_and_backend():
2424
"""Auto-detect accelerator device and distributed backend."""
2525
device_type = str(torch.accelerator.current_accelerator())
26-
backend_map = {"cuda": "nccl", "xpu": "ccl"}
26+
backend_map = {"cuda": "nccl", "xpu": "xccl"}
2727
backend = backend_map.get(device_type, "gloo")
2828
return device_type, backend
2929

0 commit comments

Comments
 (0)