diff --git a/paddle/fluid/operators/distributed/parameter_recv.cc b/paddle/fluid/operators/distributed/parameter_recv.cc index 27908aa468347d0171e6e46bc5903ad47abcde70..e9d34b173942b1ae27be3515cd9862f0b5984af0 100644 --- a/paddle/fluid/operators/distributed/parameter_recv.cc +++ b/paddle/fluid/operators/distributed/parameter_recv.cc @@ -121,11 +121,15 @@ void ParameterRecv::operator()(const RpcContext &rpc_ctx, 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; - VLOG(3) << "ParameterRecv out" << rpc_ctx.var_name; + VLOG(3) << "ParameterRecv out " << rpc_ctx.var_name; } template struct ParameterRecv;