You need to sign in or sign up before continuing.
未验证 提交 3863c6a9 编写于 作者: W Wu Yi 提交者: GitHub

Merge pull request #10126 from typhoonzero/fix_para_exe_dist_train

Fix para exe dist train
...@@ -34,7 +34,10 @@ void SendOpHandle::RunImpl() { ...@@ -34,7 +34,10 @@ void SendOpHandle::RunImpl() {
} }
in->generated_op_->Wait(dev_ctxes_[p]); in->generated_op_->Wait(dev_ctxes_[p]);
} }
this->RunAndRecordEvent([&] { op_->Run(*local_scope_, place_); }); auto &tmp_scope = local_scope_->FindVar(kLocalExecScopeName)->Get<Scope *>();
// FIXME(wuyi): can not use RunAndRecordEvent here, for it will cause dead
// lock.
op_->Run(*tmp_scope, place_);
} }
std::string SendOpHandle::Name() const { return "send"; } std::string SendOpHandle::Name() const { return "send"; }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册