From 7e714e2bdc7731a5724d9796c7d229c2acae3055 Mon Sep 17 00:00:00 2001
From: zhupy <1165938320@qq.com>
Date: Thu, 13 Jun 2019 08:03:16 +0000
Subject: [PATCH] fix split arm kernel

---
 paddle/fluid/lite/arm/math/split.cc            | 4 ++--
 paddle/fluid/lite/arm/math/split.h             | 2 +-
 paddle/fluid/lite/kernels/arm/split_compute.cc | 2 +-
 paddle/fluid/lite/operators/op_params.h        | 4 ++--
 4 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/paddle/fluid/lite/arm/math/split.cc b/paddle/fluid/lite/arm/math/split.cc
index 6dd6de6242e..bf8d50590ff 100644
--- a/paddle/fluid/lite/arm/math/split.cc
+++ b/paddle/fluid/lite/arm/math/split.cc
@@ -52,10 +52,10 @@ void split_cpy<float>(const float* din, float* dout, int num) {
 }
 
 template <>
-void split<float>(const float* din, std::vector<lite::Tensor*>* dout,
+void split<float>(const float* din, const std::vector<lite::Tensor*>& dout,
                   const int axis, const std::vector<int>& in_strides) {
   int input_offset = 0;
-  for (auto out : *dout) {
+  for (auto out : dout) {
     auto out_dim = out->dims();
     std::vector<int> out_strides(out_dim.size());
     out_strides[out_dim.size() - 1] = out_dim[out_dim.size() - 1];
diff --git a/paddle/fluid/lite/arm/math/split.h b/paddle/fluid/lite/arm/math/split.h
index 9b5651d81ff..643214e174c 100644
--- a/paddle/fluid/lite/arm/math/split.h
+++ b/paddle/fluid/lite/arm/math/split.h
@@ -26,7 +26,7 @@ template <typename T>
 void split_cpy(const T* din, T* dout, int num);
 
 template <typename T>
-void split(const T* din, std::vector<lite::Tensor*>* dout, const int axis,
+void split(const T* din, const std::vector<lite::Tensor*>& dout, const int axis,
            const std::vector<int>& in_strides);
 
 }  // namespace math
diff --git a/paddle/fluid/lite/kernels/arm/split_compute.cc b/paddle/fluid/lite/kernels/arm/split_compute.cc
index 9da69894592..3c2416bd690 100644
--- a/paddle/fluid/lite/kernels/arm/split_compute.cc
+++ b/paddle/fluid/lite/kernels/arm/split_compute.cc
@@ -24,7 +24,7 @@ namespace arm {
 void SplitCompute::Run() {
   auto& param = Param<operators::SplitParam>();
   const float* din = param.x->data<float>();
-  auto* dout = param.output;
+  auto& dout = param.output;
   auto in_dim = param.x->dims();
   std::vector<int> in_strides(in_dim.size());
   in_strides[in_dim.size() - 1] = in_dim[in_dim.size() - 1];
diff --git a/paddle/fluid/lite/operators/op_params.h b/paddle/fluid/lite/operators/op_params.h
index eee0d90dba2..9986d199393 100644
--- a/paddle/fluid/lite/operators/op_params.h
+++ b/paddle/fluid/lite/operators/op_params.h
@@ -177,10 +177,10 @@ struct DropoutParam {
 // For Split op
 struct SplitParam {
   lite::Tensor* x{};
-  std::vector<lite::Tensor*>* output{};
+  std::vector<lite::Tensor*> output{};
   int axis{-1};
   int num{0};
-  std::vector<int>* sections;
+  std::vector<int> sections;
 };
 
 /// ----------------------- element wise operators ----------------------
-- 
GitLab