diff --git a/lite/backends/arm/math/gemv_arm_int8.cc b/lite/backends/arm/math/gemv_arm_int8.cc index c7effde5b64..a68a75b7edf 100644 --- 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 index 5106f3c7f12..cf484139eed 100644 --- a/lite/kernels/arm/matmul_compute.cc +++ b/lite/kernels/arm/matmul_compute.cc @@ -29,7 +29,13 @@ void MatMulCompute::PrepareForRun() { } template <> -void MatMulCompute::PrepareForRun() { +void MatMulCompute::PrepareForRun() {} + +template <> +void MatMulCompute::ReInitWhenNeeded() {} + +template <> +void MatMulCompute::ReInitWhenNeeded() { auto& ctx = this->ctx_->template As(); auto& param = Param(); auto x_dims = param.X->dims(); diff --git a/lite/kernels/arm/matmul_compute.h b/lite/kernels/arm/matmul_compute.h index 865a3f2c717..b795bb84868 100644 --- a/lite/kernels/arm/matmul_compute.h +++ b/lite/kernels/arm/matmul_compute.h @@ -30,6 +30,8 @@ class MatMulCompute : public KernelLite { void PrepareForRun() override; + void ReInitWhenNeeded() override; + void Run() override; virtual ~MatMulCompute() = default;