提交 8ee83725 编写于 作者: T typhoonzero

fix send op handle local scope

上级 d15b264e
......@@ -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"; }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册