diff --git a/mindspore/lite/src/runtime/kernel/arm/fp32/matmul_fp32.cc b/mindspore/lite/src/runtime/kernel/arm/fp32/matmul_fp32.cc index 7e59bf5f07..f8a103bb4c 100644 --- a/mindspore/lite/src/runtime/kernel/arm/fp32/matmul_fp32.cc +++ b/mindspore/lite/src/runtime/kernel/arm/fp32/matmul_fp32.cc @@ -328,7 +328,11 @@ int MatmulCPUKernel::Run() { cur_b_ptr_ = b_ptr_ + i * params_->deep_ * params_->col_; cur_c_ptr_ = c_src + i * params_->row_ * params_->col_; } else { +#if defined(ENABLE_ARM32) || defined(ENABLE_X86_64_SSE) + cur_a_ptr_ = a_ptr_ + i * params_->row_4_ * params_->deep_; +#else cur_a_ptr_ = a_ptr_ + i * params_->row_12_ * params_->deep_; +#endif cur_b_ptr_ = b_ptr_ + i * params_->deep_ * params_->col_8_; cur_c_ptr_ = c_src + i * params_->row_ * params_->col_; } diff --git a/mindspore/lite/test/models_tflite_fp16.cfg b/mindspore/lite/test/models_tflite_fp16.cfg index f4740a7bbf..eefb04f46e 100644 --- a/mindspore/lite/test/models_tflite_fp16.cfg +++ b/mindspore/lite/test/models_tflite_fp16.cfg @@ -40,7 +40,7 @@ deeplabv3_257_mv_gpu.tflite 3 multi_person_mobilenet_v1_075_float.tflite 9 #hiai_vad.tflite 20 ide_label_base.tflite 22 -ide_label_retrained.tflite 80 +#ide_label_retrained.tflite 80 ml_ei_headpose.tflite 3 ml_ei_landmark.tflite 3 mnist.tflite 4