Skip to content

Commit 8ee8372

Browse files
committed
fix send op handle local scope
1 parent d15b264 commit 8ee8372

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

paddle/fluid/framework/details/send_op_handle.cc

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,9 @@ void SendOpHandle::RunImpl() {
3434
}
3535
in->generated_op_->Wait(dev_ctxes_[p]);
3636
}
37-
this->RunAndRecordEvent([&] { op_->Run(*local_scope_, place_); });
37+
auto &tmp_scope = local_scope_->FindVar(kLocalExecScopeName)->Get<Scope *>();
38+
// auto &lod_tensor = tmp_scope->FindVar(var_name)->Get<LoDTensor>();
39+
this->RunAndRecordEvent([&] { op_->Run(*tmp_scope, place_); });
3840
}
3941

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

0 commit comments

Comments
 (0)