updates follow up para exe

wangkuiyi-patch-2
typhoonzero 7 years ago
parent 8ee837255e
commit 2b06b4b4e9

@ -35,8 +35,9 @@ void SendOpHandle::RunImpl() {
in->generated_op_->Wait(dev_ctxes_[p]);
}
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_); });
// 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"; }

Loading…
Cancel
Save