code optimize

6 years ago · fab8457e6b
parent ada43e89c3
commit fab8457e6b
1 changed files with 7 additions and 9 deletions
--- 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();
  }