提交 90f98016 编写于 作者: Y Yu Yang

Do not wait computation stream

上级 7ac969b8
......@@ -782,10 +782,6 @@ void ParallelExecutor::Run(const std::vector<std::string> &fetch_tensors,
}
}
for (auto &p : member_->places_) {
platform::DeviceContextPool::Instance().Get(p)->Wait();
}
for (auto &fetch_op : fetch_ops) {
fetch_op.WaitAndMergeCPUTensors();
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册