Revert "fix lookuptable in reduce strategy"

This reverts commit 0e722c5
fix_recordio_link
tangwei12 6 years ago
parent 2761eafb92
commit b58957d9d7

@ -680,8 +680,7 @@ int MultiDevSSAGraphBuilder::CreateDistTrainOp(ir::Graph *result,
} }
if (node->Op()->Type() == "split_byref" || if (node->Op()->Type() == "split_byref" ||
node->Op()->Type() == "split_selected_rows" || node->Op()->Type() == "split_selected_rows") {
node->Op()->Type() == "split_ids") {
// TODO(paddle-dev): getting the first var is not safe. // TODO(paddle-dev): getting the first var is not safe.
op_dev_id = GetVarDeviceID(*result, input_var_names[0]); op_dev_id = GetVarDeviceID(*result, input_var_names[0]);
if (strategy_.reduce_ == BuildStrategy::ReduceStrategy::kAllReduce) { if (strategy_.reduce_ == BuildStrategy::ReduceStrategy::kAllReduce) {

@ -69,12 +69,6 @@ bool IsDistTrainOp(ir::Node *node, const std::vector<std::string> &send_vars,
std::find(rpc_vars.begin(), rpc_vars.end(), var) != rpc_vars.end()) { std::find(rpc_vars.begin(), rpc_vars.end(), var) != rpc_vars.end()) {
return true; 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; return false;
}; };

Loading…
Cancel
Save