Skip to content

Commit

Permalink
[ARM]fix armv7 gemm c8 bugs (#8628)
Browse files Browse the repository at this point in the history
  • Loading branch information
xingjing1 authored Mar 15, 2022
1 parent e98503f commit d92ab45
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions lite/backends/arm/math/fp16/gemm_c8_fp16.cc
Original file line number Diff line number Diff line change
Expand Up @@ -1357,7 +1357,7 @@ void gemm_prepack_c8_fp16_small(int M,
"vmla.f16 q9, q7, d2[3]\n"
"vmla.f16 q10, q7, d4[3]\n"
"vmla.f16 q11, q7, d6[3]\n"
"sub %[cnt], #1\n"
"subs %[cnt], #1\n"
"add %[b_ptr], %[ldb]\n"
"vmla.f16 q12, q7, d1[3]\n"
"vmla.f16 q13, q7, d3[3]\n"
Expand Down Expand Up @@ -1445,7 +1445,7 @@ void gemm_prepack_c8_fp16_small(int M,
"vmla.f16 q9, q6, d3[2]\n"
"vmla.f16 q10, q6, d5[2]\n"
"vmla.f16 q11, q6, d7[2]\n"
"sub %[cnt], #1\n"
"subs %[cnt], #1\n"
"add %[b_ptr], %[ldb]\n"

"vmla.f16 q8, q7, d1[3]\n"
Expand Down Expand Up @@ -1499,7 +1499,7 @@ void gemm_prepack_c8_fp16_small(int M,
"add %[b_ptr], %[ldb]\n"
"vld1.16 {d12-d15}, [%[a_ptr]]!\n"

"sub %[cnt], #1\n"
"subs %[cnt], #1\n"
"vmla.f16 q8, q4, d1[0]\n"
"vmla.f16 q9, q5, d1[1]\n"
"vmla.f16 q10, q6, d1[2]\n"
Expand Down

0 comments on commit d92ab45

Please sign in to comment.