From 3f6ec900605d6b33d3f7f8a24ba0af95b3267153 Mon Sep 17 00:00:00 2001
From: minqiyang <minqiyang@baidu.com>
Date: Fri, 12 Oct 2018 11:26:54 +0800
Subject: [PATCH] Polish code

test=develop
---
 paddle/fluid/operators/math/selected_rows_functor.cc | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/paddle/fluid/operators/math/selected_rows_functor.cc b/paddle/fluid/operators/math/selected_rows_functor.cc
index 43d593710c2..6810c24227e 100644
--- a/paddle/fluid/operators/math/selected_rows_functor.cc
+++ b/paddle/fluid/operators/math/selected_rows_functor.cc
@@ -172,7 +172,6 @@ struct SelectedRowsSumTo<platform::CPUDeviceContext, T> {
     }
     input2->set_rows(in2_rows);
 
-    // start = std::chrono::system_clock::now();
     auto* in2_value = input2->mutable_value();
     auto* in2_data = in2_value->data<T>();
     auto blas = math::GetBlas<platform::CPUDeviceContext, T>(context);
@@ -275,7 +274,7 @@ struct MergeAdd<platform::CPUDeviceContext, T> {
     auto* input_data = input.value().data<T>();
 
     for (size_t i = 0; i < input_rows.size(); i++) {
-      size_t out_i = FindPos(merge_rows, input_rows[i]);
+      size_t out_i = rows_pos_map[input_rows[i]];
       for (int64_t j = 0; j < input_width; j++) {
         out_data[out_i * input_width + j] += input_data[i * input_width + j];
       }
-- 
GitLab