[XCCL] Use xccl for xpu backend since xccl is ready in latest PyTorch. (#3954)

This commit is contained in:
Chen Shengzhi
2025-03-04 20:05:56 +08:00
committed by GitHub
parent 19120f71f3
commit 61261b3996

View File

@@ -263,9 +263,7 @@ class ModelRunner:
if self.device == "cuda":
backend = "nccl"
elif self.device == "xpu":
# TODO(liangan1): Just use gloo to bypass the initilization fail
# Need to use xccl for xpu backend in the future
backend = "gloo"
backend = "xccl"
elif self.device == "hpu":
backend = "hccl"
elif self.device == "cpu":