From 12c5b1fea4906519bfa5ffa6167515b3c2d45067 Mon Sep 17 00:00:00 2001
From: Zhang Ting <zhangting_2017@163.com>
Date: Wed, 12 Jan 2022 16:35:36 +0800
Subject: [PATCH] [part 6]change type of function args (#38891)

---
 .../kernel_primitives/functor_primitives.h    | 30 +++++++++----------
 paddle/pten/kernels/gpu/math_kernel.cu        |  2 +-
 2 files changed, 16 insertions(+), 16 deletions(-)

diff --git a/paddle/fluid/operators/kernel_primitives/functor_primitives.h b/paddle/fluid/operators/kernel_primitives/functor_primitives.h
index 2bd8721b82f..5e3c1fc202d 100644
--- a/paddle/fluid/operators/kernel_primitives/functor_primitives.h
+++ b/paddle/fluid/operators/kernel_primitives/functor_primitives.h
@@ -53,7 +53,7 @@ struct ExpFunctor {
 
   HOSTDEVICE explicit inline ExpFunctor(int n) {}
 
-  HOSTDEVICE inline Ty operator()(const Tx& x) const {
+  HOSTDEVICE inline Ty operator()(const Tx x) const {
     return static_cast<Ty>(details::Exp(x));
   }
 };
@@ -67,7 +67,7 @@ struct IdentityFunctor {
 
   HOSTDEVICE explicit inline IdentityFunctor(int n) {}
 
-  HOSTDEVICE inline Ty operator()(const Tx& x) const {
+  HOSTDEVICE inline Ty operator()(const Tx x) const {
     return static_cast<Ty>(x);
   }
 };
@@ -85,7 +85,7 @@ struct DivideFunctor {
 
   HOSTDEVICE explicit inline DivideFunctor(int n) : n_inv((MPType)(1.0 / n)) {}
 
-  HOSTDEVICE inline Ty operator()(const Tx& x) const {
+  HOSTDEVICE inline Ty operator()(const Tx x) const {
     return static_cast<Ty>(static_cast<MPType>(x) * n_inv);
   }
 
@@ -102,7 +102,7 @@ struct InverseFunctor {
 
   HOSTDEVICE explicit inline InverseFunctor(int n) {}
 
-  HOSTDEVICE inline Ty operator()(const Tx& x) const {
+  HOSTDEVICE inline Ty operator()(const Tx x) const {
     return static_cast<Ty>(-x);
   }
 };
@@ -116,7 +116,7 @@ struct SquareFunctor {
 
   HOSTDEVICE explicit inline SquareFunctor(int n) {}
 
-  HOSTDEVICE inline Ty operator()(const Tx& x) const {
+  HOSTDEVICE inline Ty operator()(const Tx x) const {
     return static_cast<Ty>(x) * static_cast<Ty>(x);
   }
 };
@@ -130,7 +130,7 @@ template <typename T>
 struct MinFunctor {
   inline T initial() { return static_cast<T>(std::numeric_limits<T>::max()); }
 
-  __device__ __forceinline__ T operator()(const T& a, const T& b) const {
+  __device__ __forceinline__ T operator()(const T a, const T b) const {
     return (b < a) ? b : a;
   }
 };
@@ -144,7 +144,7 @@ struct MaxFunctor {
     return static_cast<T>(std::numeric_limits<T>::lowest());
   }
 
-  __device__ __forceinline__ T operator()(const T& a, const T& b) const {
+  __device__ __forceinline__ T operator()(const T a, const T b) const {
     return (b > a) ? b : a;
   }
 };
@@ -156,7 +156,7 @@ template <typename T>
 struct AddFunctor {
   inline T initial() { return static_cast<T>(0.0f); }
 
-  __device__ __forceinline__ T operator()(const T& a, const T& b) const {
+  __device__ __forceinline__ T operator()(const T a, const T b) const {
     return b + a;
   }
 };
@@ -168,7 +168,7 @@ template <typename T>
 struct MulFunctor {
   inline T initial() { return static_cast<T>(1.0f); }
 
-  __device__ __forceinline__ T operator()(const T& a, const T& b) const {
+  __device__ __forceinline__ T operator()(const T a, const T b) const {
     return b * a;
   }
 };
@@ -180,7 +180,7 @@ template <typename T>
 struct LogicalOrFunctor {
   inline T initial() { return static_cast<T>(false); }
 
-  __device__ __forceinline__ T operator()(const T& a, const T& b) const {
+  __device__ __forceinline__ T operator()(const T a, const T b) const {
     return b || a;
   }
 };
@@ -192,7 +192,7 @@ template <typename T>
 struct LogicalAndFunctor {
   inline T initial() { return static_cast<T>(true); }
 
-  __device__ __forceinline__ T operator()(const T& a, const T& b) const {
+  __device__ __forceinline__ T operator()(const T a, const T b) const {
     return b && a;
   }
 };
@@ -204,7 +204,7 @@ template <typename T>
 struct SubFunctor {
   inline T initial() { return static_cast<T>(0.0f); }
 
-  inline HOSTDEVICE T operator()(const T& a, const T& b) const { return a - b; }
+  inline HOSTDEVICE T operator()(const T a, const T b) const { return a - b; }
 };
 
 /**
@@ -214,7 +214,7 @@ template <typename T, typename Enable = void>
 struct DivFunctor {
   inline T initial() { return static_cast<T>(1.0f); }
 
-  inline HOSTDEVICE T operator()(const T& a, const T& b) const { return a / b; }
+  inline HOSTDEVICE T operator()(const T a, const T b) const { return a / b; }
 };
 
 template <typename T>
@@ -222,7 +222,7 @@ struct DivFunctor<T,
                   typename std::enable_if<std::is_integral<T>::value>::type> {
   inline T initial() { return static_cast<T>(1.0f); }
 
-  inline HOSTDEVICE T operator()(const T& a, const T& b) const {
+  inline HOSTDEVICE T operator()(const T a, const T b) const {
     // For int32/int64, need to check whether the divison is zero.
     PADDLE_ENFORCE_NE(b, 0,
                       platform::errors::InvalidArgument(
@@ -239,7 +239,7 @@ template <typename T>
 struct FloorDivFunctor {
   inline T initial() { return static_cast<T>(1.0f); }
 
-  inline HOSTDEVICE T operator()(const T& a, const T& b) const {
+  inline HOSTDEVICE T operator()(const T a, const T b) const {
     PADDLE_ENFORCE_NE(b, 0,
                       platform::errors::InvalidArgument(
                           "Integer division by zero encountered "
diff --git a/paddle/pten/kernels/gpu/math_kernel.cu b/paddle/pten/kernels/gpu/math_kernel.cu
index f41934313d6..55708063803 100644
--- a/paddle/pten/kernels/gpu/math_kernel.cu
+++ b/paddle/pten/kernels/gpu/math_kernel.cu
@@ -61,7 +61,7 @@ struct DivideFunctor {
   HOSTDEVICE explicit inline DivideFunctor(int n)
       : n_inv(static_cast<T>(1.0 / n)) {}
 
-  HOSTDEVICE inline T operator()(const T& x) const { return x * n_inv; }
+  HOSTDEVICE inline T operator()(const T x) const { return x * n_inv; }
 
  private:
   T n_inv;
-- 
GitLab