code optimize

fab8457e · Qiao Longfei · ada43e89 · fab8457e
隐藏空白更改
内联并排

Showing with 7 addition and 9 deletion

paddle/fluid/framework/details/async_ssa_graph_executor.cc paddle/fluid/framework/details/async_ssa_graph_executor.cc +7 -9

未找到文件。
--- a/paddle/fluid/framework/details/async_ssa_graph_executor.cc
+++ b/paddle/fluid/framework/details/async_ssa_graph_executor.cc
@@ -68,20 +68,18 @@ FeedFetchList AsyncSSAGraphExecutor::Run(

    if (pool_) {
      run_futures.emplace_back(pool_->enqueue(std::move(call)));
+      for (auto &f : run_futures) {
+        if (exception_holder_.IsCaught()) {
+          f.wait();
+        } else {
+          fetch_data.emplace_back(std::move(f.get()));
+        }
+      }
    } else {
      fetch_data.emplace_back(std::move(call()));
    }
  }

-  if (pool_) {
-    for (auto &f : run_futures) {
-      if (exception_holder_.IsCaught()) {
-        f.wait();
-      } else {
-        fetch_data.emplace_back(std::move(f.get()));
-      }
-    }
-  }
  if (exception_holder_.IsCaught()) {
    exception_holder_.ReThrow();
  }