fix select_rows mergeadd bug, test=develop (#20876)

6 years ago · d428912503
parent c51722c820
commit d428912503
1 changed files with 1 additions and 1 deletions
--- 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,