提交 0997cf8f 编写于 作者: Q Qiao Longfei

add more check

上级 93464b25
...@@ -121,11 +121,15 @@ void ParameterRecv<T>::operator()(const RpcContext &rpc_ctx, ...@@ -121,11 +121,15 @@ void ParameterRecv<T>::operator()(const RpcContext &rpc_ctx,
PADDLE_THROW("unsupported recieved var type"); PADDLE_THROW("unsupported recieved var type");
} }
} }
PADDLE_ENFORCE_EQ(recv_numel, recv_tensor->numel()); auto numel = recv_tensor->numel();
if (recv_numel != numel) {
LOG(FATAL) << "recv_numel: " << recv_numel << " acture numel: " << numel;
}
PADDLE_ENFORCE_EQ(recv_numel, numel);
} }
delete local_scope; delete local_scope;
VLOG(3) << "ParameterRecv out" << rpc_ctx.var_name; VLOG(3) << "ParameterRecv out " << rpc_ctx.var_name;
} }
template struct ParameterRecv<float>; template struct ParameterRecv<float>;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册