Unverified Commit a8134aef authored by Xinyu Chen's avatar Xinyu Chen Committed by GitHub
Browse files

[XPU] check is_xccl_available before oneccl warmup (#39302)


Signed-off-by: default avatarXinyu Chen <xinyu1.chen@intel.com>
parent 2800706f
...@@ -86,7 +86,8 @@ class XPUWorker(Worker): ...@@ -86,7 +86,8 @@ class XPUWorker(Worker):
) )
# global all_reduce needed for overall oneccl warm up # global all_reduce needed for overall oneccl warm up
torch.distributed.all_reduce(torch.zeros(1).xpu()) if torch.distributed.is_xccl_available():
torch.distributed.all_reduce(torch.zeros(1).xpu())
# Set random seed. # Set random seed.
set_random_seed(self.model_config.seed) set_random_seed(self.model_config.seed)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment