remove wait if no fetch (#36150)

b3d2dc7b · Zeng Jinle · GitHub · 667bf188 · b3d2dc7b
隐藏空白更改
内联并排

Showing with 5 addition and 3 deletion

paddle/fluid/framework/details/fast_threaded_ssa_graph_executor.cc ...uid/framework/details/fast_threaded_ssa_graph_executor.cc +5 -3

未找到文件。
--- a/paddle/fluid/framework/details/fast_threaded_ssa_graph_executor.cc
+++ b/paddle/fluid/framework/details/fast_threaded_ssa_graph_executor.cc
@@ -139,10 +139,12 @@ FetchResultType FastThreadedSSAGraphExecutor::Run(
    }
  }
  // Wait FetchOps.
-  ClearFetchOp(graph_, &fetch_ops);
+  if (!fetch_ops.empty()) {
+    ClearFetchOp(graph_, &fetch_ops);
-  for (auto &place : places_) {
+    for (auto &place : places_) {
-    fetch_ctxs_.Get(place)->Wait();
+      fetch_ctxs_.Get(place)->Wait();
+    }
  }
  return fetches;