add more check

mixed_precision_init
Qiao Longfei 6 years ago
parent 93464b25ac
commit 0997cf8f65

@ -121,7 +121,11 @@ 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;

Loading…
Cancel
Save