Throw error

5a02739c · Yu Yang · f385228f · 5a02739c
隐藏空白更改
内联并排

Showing with 5 addition and 0 deletion

paddle/fluid/framework/details/threaded_ssa_graph_executor.cc ...le/fluid/framework/details/threaded_ssa_graph_executor.cc +5 -0

未找到文件。
--- 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);
      }