未验证 提交 6b4b9fea 编写于 作者: B Baibaifan 提交者: GitHub

hccl init sync (#34918)

上级 49818943
......@@ -87,6 +87,8 @@ class CCommInitOpAscend : public framework::OperatorBase {
}
PADDLE_ENFORCE_NPU_SUCCESS(platform::dynload::HcclBroadcast(
buff, size, HCCL_DATA_TYPE_FP32, 0, comm->comm(), stream));
// Synchronize stream to find hccl error in time.
PADDLE_ENFORCE_NPU_SUCCESS(aclrtSynchronizeStream(stream));
VLOG(3) << "Build connection successful.";
#else
PADDLE_THROW(platform::errors::PreconditionNotMet(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册