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

pull/9019/head
yangruoqi713 4 years ago
parent 4b907fdc20
commit e50a9cfd1c

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