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,11 +139,13 @@ FetchResultType FastThreadedSSAGraphExecutor::Run(
    }
  }
  // Wait FetchOps.
+  if (!fetch_ops.empty()) {
    ClearFetchOp(graph_, &fetch_ops);

    for (auto &place : places_) {
      fetch_ctxs_.Get(place)->Wait();
    }
+  }

  return fetches;
 }