Throw error

7 years ago · 5a02739ce9
parent f385228f05
commit 5a02739ce9
1 changed files with 5 additions and 0 deletions
--- a/paddle/fluid/framework/details/threaded_ssa_graph_executor.cc
+++ b/paddle/fluid/framework/details/threaded_ssa_graph_executor.cc
@ -208,6 +208,11 @@ void ThreadedSSAGraphExecutor::RunOp(
    try {
      VLOG(10) << op->DebugString();
      op->Run(use_event_);
+
+      for (auto &dev_ctx : op->dev_ctx_) {
+        dev_ctx.second->Wait();  // Sync error
+      }
+
      for (auto *ready : *ready_buffer) {
        ready->store(true, std::memory_order_release);
      }