未验证 提交 5d638feb 编写于 作者: H houj04 提交者: GitHub

[XPU] do not call check_nccl_version_for_p2p under xpu (#53862)

* [XPU] do not call check_nccl_version_for_p2p under xpu

* refine code.
上级 d294eef9
...@@ -189,6 +189,7 @@ class HybridCommunicateGroup: ...@@ -189,6 +189,7 @@ class HybridCommunicateGroup:
# create p2p_groups # create p2p_groups
if self._pp_degree > 1: if self._pp_degree > 1:
if paddle.framework.core.is_compiled_with_nccl():
check_nccl_version_for_p2p() check_nccl_version_for_p2p()
self._set_p2p_group() self._set_p2p_group()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册