From b58957d9d792b8ec85ad460a02ecc1f13575e7cd Mon Sep 17 00:00:00 2001 From: tangwei12 Date: Thu, 25 Oct 2018 17:08:40 +0800 Subject: [PATCH] Revert "fix lookuptable in reduce strategy" This reverts commit 0e722c5 --- paddle/fluid/framework/details/multi_devices_graph_pass.cc | 3 +-- paddle/fluid/framework/ir/graph.cc | 6 ------ 2 files changed, 1 insertion(+), 8 deletions(-) diff --git a/paddle/fluid/framework/details/multi_devices_graph_pass.cc b/paddle/fluid/framework/details/multi_devices_graph_pass.cc index 4f481db06..134fcee82 100644 --- a/paddle/fluid/framework/details/multi_devices_graph_pass.cc +++ b/paddle/fluid/framework/details/multi_devices_graph_pass.cc @@ -680,8 +680,7 @@ int MultiDevSSAGraphBuilder::CreateDistTrainOp(ir::Graph *result, } if (node->Op()->Type() == "split_byref" || - node->Op()->Type() == "split_selected_rows" || - node->Op()->Type() == "split_ids") { + node->Op()->Type() == "split_selected_rows") { // TODO(paddle-dev): getting the first var is not safe. op_dev_id = GetVarDeviceID(*result, input_var_names[0]); if (strategy_.reduce_ == BuildStrategy::ReduceStrategy::kAllReduce) { diff --git a/paddle/fluid/framework/ir/graph.cc b/paddle/fluid/framework/ir/graph.cc index 87fc5e689..398f70959 100644 --- a/paddle/fluid/framework/ir/graph.cc +++ b/paddle/fluid/framework/ir/graph.cc @@ -69,12 +69,6 @@ bool IsDistTrainOp(ir::Node *node, const std::vector &send_vars, std::find(rpc_vars.begin(), rpc_vars.end(), var) != rpc_vars.end()) { return true; } - - if (!(var.find(".block") == std::string::npos && - var.find(".pserver") != std::string::npos) && - std::find(rpc_vars.begin(), rpc_vars.end(), var) != rpc_vars.end()) { - return true; - } } return false; }; -- GitLab