Fix bug

7 years ago · fbbcedda01
parent 7643c2cbab
commit fbbcedda01
1 changed files with 1 additions and 2 deletions
--- a/paddle/fluid/framework/parallel_executor.cc
+++ b/paddle/fluid/framework/parallel_executor.cc
@ -108,14 +108,13 @@ struct OpHandle {
  }

  virtual void Wait(platform::DeviceContext *waited_dev) {
-    if (platform::is_cpu_place(waited_dev->GetPlace()) && events_.empty()) {
+    if (platform::is_cpu_place(waited_dev->GetPlace()) || events_.empty()) {
      for (auto &dev_ctx : dev_ctx_) {
        dev_ctx.second->Wait();
      }
    } else {
      auto stream =
          static_cast<platform::CUDADeviceContext *>(waited_dev)->stream();
-
      for (auto &ev : events_) {
        PADDLE_ENFORCE(cudaStreamWaitEvent(stream, ev.second, 0));
      }