Skip to content

P
Paddle

机器未来 / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle
未验证 提交 6b4b9fea 编写于 作者: B Baibaifan 提交者： GitHub
浏览文件
操作

hccl init sync (#34918)

上级 49818943
隐藏空白更改
内联 并排
Showing with 2 addition and 0 deletion
paddle/fluid/operators/collective/c_comm_init_hccl_op.cc
浏览文件 @ 6b4b9fea
...@@ -87,6 +87,8 @@ class CCommInitOpAscend : public framework::OperatorBase { ...@@ -87,6 +87,8 @@ class CCommInitOpAscend : public framework::OperatorBase {
} }
PADDLE_ENFORCE_NPU_SUCCESS(platform::dynload::HcclBroadcast( PADDLE_ENFORCE_NPU_SUCCESS(platform::dynload::HcclBroadcast(
buff, size, HCCL_DATA_TYPE_FP32, 0, comm->comm(), stream)); 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."; VLOG(3) << "Build connection successful.";
#else #else
PADDLE_THROW(platform::errors::PreconditionNotMet( PADDLE_THROW(platform::errors::PreconditionNotMet(
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或