From f52714d391d49230e0cfc630a5fcbb35c06c941a Mon Sep 17 00:00:00 2001
From: Yu Yang <yuyang18@baidu.com>
Date: Mon, 19 Mar 2018 16:33:35 +0800
Subject: [PATCH] Debug

---
 paddle/fluid/framework/parallel_executor.cc | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/paddle/fluid/framework/parallel_executor.cc b/paddle/fluid/framework/parallel_executor.cc
index c8dd3f9151d..1e1a5477a03 100644
--- a/paddle/fluid/framework/parallel_executor.cc
+++ b/paddle/fluid/framework/parallel_executor.cc
@@ -172,6 +172,10 @@ struct FetchOpHandle : public OpHandle {
   }
 
   void Run() override {
+    for (auto *input : inputs_) {
+      input->generated_op_->Wait(nullptr);
+    }
+
     tensors_.resize(inputs_.size());
     auto *var = static_cast<VarHandle *>(inputs_[0]);
     auto &var_name = var->name_;
-- 
GitLab