!9019 [MSLITE][Develop] fix bug of arm32 cpu fp32 matmul

From: @yangruoqi713
Reviewed-by: @HilbertDavid,@ddwsky
Signed-off-by: @HilbertDavid
pull/9019/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit 4d9d5852aa

@ -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_;
}

@ -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

Loading…
Cancel
Save