Do not wait computation stream

7 years ago · 90f980167d
parent 7ac969b88c
commit 90f980167d
1 changed files with 0 additions and 4 deletions
--- a/paddle/fluid/framework/parallel_executor.cc
+++ b/paddle/fluid/framework/parallel_executor.cc
@ -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();
  }