fix bug vtanh

paddle/fluid/operators/math/jit_kernel.h

@@ -29,7 +29,6 @@ namespace jitkernel {
 #define SIGMOID_THRESHOLD_MIN -40.0
 #define SIGMOID_THRESHOLD_MAX 13.0
 #define EXP_MAX_INPUT 40.0
-
 #define AVX_FLOAT_BLOCK 8
 #define AVX2_FLOAT_BLOCK 8
 #define AVX512_FLOAT_BLOCK 16
@@ -40,8 +39,9 @@ class Kernel {
  public:
   Kernel() = default;
   virtual ~Kernel() = default;
-
+  int num_{0};
- private:
+  int end_{0};
+  int rest_{0};
   DISABLE_COPY_AND_ASSIGN(Kernel);
 };
 
@@ -95,13 +95,13 @@ class VExpKernel : public Kernel {
 template <typename T>
 class VSigmoidKernel : public Kernel {
  public:
-  virtual void Compute(const int n, const T *x, T *y) const = 0;
+  virtual void Compute(const T *x, T *y) const = 0;
 };
 
 template <typename T>
 class VTanhKernel : public Kernel {
  public:
-  virtual void Compute(const int n, const T *x, T *y) const = 0;
+  virtual void Compute(const T *x, T *y) const = 0;
 };
 
 template <typename T>

paddle/fluid/operators/math/jit_kernel_test.cc

@@ -195,7 +195,7 @@ TEST(JitKernel, vsigmoid) {
     auto trefe = GetCurrentUS();
     auto ttgts = GetCurrentUS();
     for (int i = 0; i < repeat; ++i) {
-      ker->Compute(d, x_data, ztgt_data);
+      ker->Compute(x_data, ztgt_data);
     }
     auto ttgte = GetCurrentUS();
 
@@ -227,7 +227,7 @@ void vtanh_better(
         vaddbias,
     const int n, const float* x, float* y) {
   vscal->Compute(n, 2.f, x, y);
-  vsigmoid->Compute(n, y, y);
+  vsigmoid->Compute(y, y);
   vscal->Compute(n, 2.f, y);
   vaddbias->Compute(n, -1.f, y, y);
 }
@@ -261,7 +261,7 @@ TEST(JitKernel, vtanh) {
     auto trefe = GetCurrentUS();
     auto ttgts = GetCurrentUS();
     for (int i = 0; i < repeat; ++i) {
-      ker->Compute(d, x_data, ztgt_data);
+      ker->Compute(x_data, ztgt_data);
     }
     auto ttgte = GetCurrentUS();