From d4289125038c96a7a6ab4ed1f7ef9d28ea83bc53 Mon Sep 17 00:00:00 2001
From: zhang wenhui <frankwhzhang@126.com>
Date: Wed, 30 Oct 2019 10:36:28 +0800
Subject: [PATCH] fix select_rows mergeadd bug, test=develop (#20876)

---
 paddle/fluid/operators/math/selected_rows_functor.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/paddle/fluid/operators/math/selected_rows_functor.cc b/paddle/fluid/operators/math/selected_rows_functor.cc
index f73c9bb9dc..d9a3c668a1 100644
--- a/paddle/fluid/operators/math/selected_rows_functor.cc
+++ b/paddle/fluid/operators/math/selected_rows_functor.cc
@@ -332,7 +332,7 @@ struct MergeAdd<platform::CPUDeviceContext, T> {
       int64_t copied_numel = 0;
       for (auto* in : inputs) {
         auto* in_data = in->value().data<T>();
-        auto in_numel = in->value().numel();
+        auto in_numel = in->rows().size() * input_width;
         memory::Copy(boost::get<platform::CPUPlace>(out_place),
                      out_data + copied_numel,
                      boost::get<platform::CPUPlace>(in_place), in_data,