From fad42fe7ccf9bc557482e09473d049b456b7466c Mon Sep 17 00:00:00 2001 From: Qiao Longfei Date: Sat, 27 Oct 2018 10:43:50 +0800 Subject: [PATCH] broadcast handle not inited parameter --- paddle/fluid/framework/details/broadcast_op_handle.cc | 4 ++++ paddle/fluid/framework/parallel_executor.cc | 4 ++++ 2 files changed, 8 insertions(+) diff --git a/paddle/fluid/framework/details/broadcast_op_handle.cc b/paddle/fluid/framework/details/broadcast_op_handle.cc index 4fdab5cd9..a8de23839 100644 --- a/paddle/fluid/framework/details/broadcast_op_handle.cc +++ b/paddle/fluid/framework/details/broadcast_op_handle.cc @@ -52,6 +52,10 @@ void BroadcastOpHandle::RunImpl() { var_scopes.at(in_var_handle->scope_idx_)->FindVar(in_var_handle->name_); PADDLE_ENFORCE_NOT_NULL(in_var); Tensor &in_tensor = VariableVisitor::GetMutableTensor(in_var); + if (!in_tensor.IsInitialized()) { + VLOG(3) << "in var " << in_var_handle->name_ << "not inited, return!"; + return; + } InitOutputValue(*in_var_handle, out_var_handles); diff --git a/paddle/fluid/framework/parallel_executor.cc b/paddle/fluid/framework/parallel_executor.cc index 7dad872dd..7c2fcdb1a 100644 --- a/paddle/fluid/framework/parallel_executor.cc +++ b/paddle/fluid/framework/parallel_executor.cc @@ -179,6 +179,10 @@ void ParallelExecutor::BCastParamsToDevices( } auto &main_tensor = main_var->Get(); + if (!main_tensor.IsInitialized()) { + VLOG(3) << "one in var not inited, return!"; + continue; + } auto &dims = main_tensor.dims(); if (paddle::platform::is_gpu_place(main_tensor.place())) { #ifdef PADDLE_WITH_CUDA -- GitLab