fix send op handle local scope

7 years ago · 8ee837255e
parent d15b264ed3
commit 8ee837255e
1 changed files with 3 additions and 1 deletions
--- a/paddle/fluid/framework/details/send_op_handle.cc
+++ b/paddle/fluid/framework/details/send_op_handle.cc
@ -34,7 +34,9 @@ void SendOpHandle::RunImpl() {
    }
    in->generated_op_->Wait(dev_ctxes_[p]);
  }
-  this->RunAndRecordEvent([&] { op_->Run(*local_scope_, place_); });
+  auto &tmp_scope = local_scope_->FindVar(kLocalExecScopeName)->Get<Scope *>();
+  // auto &lod_tensor = tmp_scope->FindVar(var_name)->Get<LoDTensor>();
+  this->RunAndRecordEvent([&] { op_->Run(*tmp_scope, place_); });
 }

 std::string SendOpHandle::Name() const { return "send"; }