diff --git a/mindspore-lite/src/litert/kernel/cpu/nnacl_c/fp32/matmul_avx_fp32.c b/mindspore-lite/src/litert/kernel/cpu/nnacl_c/fp32/matmul_avx_fp32.c index 7f49814ee702eaf24f33d083de113417fabd5771..4988391eb6477a4d04746259a0c0a0098e388155 100644 --- a/mindspore-lite/src/litert/kernel/cpu/nnacl_c/fp32/matmul_avx_fp32.c +++ b/mindspore-lite/src/litert/kernel/cpu/nnacl_c/fp32/matmul_avx_fp32.c @@ -19,6 +19,9 @@ void MatVecMulAvxFp32(const float *a, const float *b, float *c, const float *bias, int act_type, int depth, int cur_col, int col_align) { // one time process 32 out_channel + if (cur_col % C8NUM != 0) { + cur_col = UP_ROUND(cur_col, C8NUM); + } int col_block = C32NUM; int act_flag = C0NUM; if (act_type == ActType_Relu6) {