From 3a46f4b9ebd06561758d93c657c2d8f9344b2772 Mon Sep 17 00:00:00 2001
From: GaoWei8 <53294385+GaoWei8@users.noreply.github.com>
Date: Thu, 13 Feb 2020 10:46:18 +0800
Subject: [PATCH] Replace GRU RowWiseAdd Eigen with c implementation (#2712)

* replace gru RowWiseAdd Eigen with c implementation
test=develop
---
 lite/backends/x86/math/math_function.cc | 18 ++++++++++--------
 1 file changed, 10 insertions(+), 8 deletions(-)

diff --git a/lite/backends/x86/math/math_function.cc b/lite/backends/x86/math/math_function.cc
index 822b7df936..f242e14ad1 100644
--- a/lite/backends/x86/math/math_function.cc
+++ b/lite/backends/x86/math/math_function.cc
@@ -110,11 +110,11 @@ void set_constant(const lite::Context<Target>& context,
                   lite::Tensor* tensor,
                   float value) {
   TensorSetConstantWithTarget<Target> func(context, tensor, value);
-  //#ifdef PADDLE_WITH_CUDA
+  // #ifdef PADDLE_WITH_CUDA
   // tensor->target().apply_visitor(func);
-  //#else
+  // #else
   func();
-  //#endif
+  // #endif
 }
 
 template <typename T>
@@ -128,12 +128,14 @@ struct RowwiseAdd<lite::TargetType::kX86, T> {
     PADDLE_ENFORCE_EQ(vector.numel(), size);
     PADDLE_ENFORCE_EQ(output->dims(), in_dims);
 
-    auto in = lite::fluid::EigenMatrix<T>::From(input);
-    auto vec = lite::fluid::EigenVector<T>::Flatten(vector);
-    auto out = lite::fluid::EigenMatrix<T>::From(*output);
-
+    const T* input_data = input.data<T>();
+    const T* vector_data = vector.data<T>();
+    T* output_data = output->mutable_data<T>();
     for (int64_t i = 0; i < in_dims[0]; ++i) {
-      out.chip(i, 0) = in.chip(i, 0) + vec;
+      for (int64_t j = 0; j < size; ++j) {
+        output_data[i * in_dims[0] + j] =
+            input_data[i * in_dims[0] + j] + vector_data[j];
+      }
     }
   }
 };
-- 
GitLab