From b64dde2a03ff9bdb501944ff51dc01359fc67833 Mon Sep 17 00:00:00 2001
From: Yang Yang <yangyang62@baidu.com>
Date: Tue, 23 Jan 2018 22:00:58 +0000
Subject: [PATCH] fix share tensor

---
 paddle/operators/parallel_do_op.cc | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/paddle/operators/parallel_do_op.cc b/paddle/operators/parallel_do_op.cc
index d977ec43ff..284623c743 100644
--- a/paddle/operators/parallel_do_op.cc
+++ b/paddle/operators/parallel_do_op.cc
@@ -247,14 +247,13 @@ class ParallelDoGradOp : public framework::OperatorBase {
       auto *tmp = sub_scopes[0]->Var(&tmp_name);
 
       for (size_t i = 1; i < sub_scopes.size(); ++i) {
-        if (!(places[i] == places[0])) {
-          CopyOrShare(*sub_scopes[i]->FindVar(s), places[0], tmp);
-          WaitOnPlace(places[0]);
-        }
+        CopyOrShare(*sub_scopes[i]->FindVar(s), places[0], tmp);
+        WaitOnPlace(places[0]);
 
         auto sum_op = framework::OpRegistry::CreateOp(
             "sum", {{"X", {s, tmp_name}}}, {{"Out", {s}}},
             framework::AttributeMap{});
+        VLOG(3) << sum_op->DebugStringEx(sub_scopes[0]);
         sum_op->Run(*sub_scopes[0], places[0]);
         WaitOnPlace(places[0]);
       }
-- 
GitLab