From 3f88fad08ce6d7800356372e7cb20a3b70cd3208 Mon Sep 17 00:00:00 2001
From: Yu Yang <yuyang18@baidu.com>
Date: Tue, 27 Mar 2018 15:30:57 +0800
Subject: [PATCH] Fix merge op

---
 paddle/fluid/framework/details/fetch_op_handle.cc | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/paddle/fluid/framework/details/fetch_op_handle.cc b/paddle/fluid/framework/details/fetch_op_handle.cc
index 9ed974151f..4fc05b3248 100644
--- a/paddle/fluid/framework/details/fetch_op_handle.cc
+++ b/paddle/fluid/framework/details/fetch_op_handle.cc
@@ -67,9 +67,9 @@ void FetchOpHandle::RunImpl() {
       tensors_[i].ShareDataWith(t);
       tensors_[i].set_lod(t.lod());
     }
-
-    this->WaitAndMergeCPUTensors();
   }
+
+  this->WaitAndMergeCPUTensors();
 }
 
 std::string FetchOpHandle::Name() const { return "Fetch"; }
-- 
GitLab