fix matmul_int8&&gemv bugs,test=develop (#5986)

PaddlePaddle · Apr 27, 2021 · f8b5ea9 · f8b5ea9
1 parent 6958847
commit f8b5ea9
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 2 deletions.
diff --git a/lite/backends/arm/math/gemv_arm_int8.cc b/lite/backends/arm/math/gemv_arm_int8.cc
@@ -1276,7 +1276,7 @@ bool gemv_int8_trans_oth(const int8_t* A,
   int cnt_4 = tail >> 2;
   int tail_4 = tail & 3;
   int stride_in = M << 3;
-#pragma omp parallel for
+  // #pragma omp parallel for
   for (int i = 0; i < cnt; i++) {
     const int8_t* in_ptr0 = data_in;
     const int8_t* in_ptr1 = in_ptr0 + M;

diff --git a/lite/kernels/arm/matmul_compute.cc b/lite/kernels/arm/matmul_compute.cc
@@ -29,7 +29,13 @@ void MatMulCompute<PRECISION(kFloat), PRECISION(kFloat)>::PrepareForRun() {
 }
 
 template <>
-void MatMulCompute<PRECISION(kInt8), PRECISION(kFloat)>::PrepareForRun() {
+void MatMulCompute<PRECISION(kInt8), PRECISION(kFloat)>::PrepareForRun() {}
+
+template <>
+void MatMulCompute<PRECISION(kFloat), PRECISION(kFloat)>::ReInitWhenNeeded() {}
+
+template <>
+void MatMulCompute<PRECISION(kInt8), PRECISION(kFloat)>::ReInitWhenNeeded() {
   auto& ctx = this->ctx_->template As<ARMContext>();
   auto& param = Param<param_t>();
   auto x_dims = param.X->dims();

diff --git a/lite/kernels/arm/matmul_compute.h b/lite/kernels/arm/matmul_compute.h
@@ -30,6 +30,8 @@ class MatMulCompute : public KernelLite<TARGET(kARM), PType> {
 
   void PrepareForRun() override;
 
+  void ReInitWhenNeeded() override;
+
   void Run() override;
 
   virtual ~MatMulCompute() = default;