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,6 +86,7 @@ class XPUWorker(Worker):
)
# global all_reduce needed for overall oneccl warm up
if torch.distributed.is_xccl_available():
torch.distributed.all_reduce(torch.zeros(1).xpu())
# Set random 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