[XPU]Bug fix for some unexpected error when use AgRs backend on XPU device. (#36593)

Signed-off-by: yisheng <yi.sheng@intel.com>
This commit is contained in:
YiSheng5
2026-03-11 17:17:46 +08:00
committed by GitHub
parent f4ae58b38b
commit c910eeb125
2 changed files with 8 additions and 5 deletions

View File

@@ -85,6 +85,9 @@ class XPUWorker(Worker):
current_platform.dist_backend,
)
# global all_reduce needed for overall oneccl warm up
torch.distributed.all_reduce(torch.zeros(1).xpu())
# Set random seed.
set_random_seed(self.model_config.seed)