diff --git a/kernel/x86_64/sbgemm_kernel_16x4_cooperlake.c b/kernel/x86_64/sbgemm_kernel_16x4_cooperlake.c index d604235c9..c257a3f60 100644 --- a/kernel/x86_64/sbgemm_kernel_16x4_cooperlake.c +++ b/kernel/x86_64/sbgemm_kernel_16x4_cooperlake.c @@ -306,9 +306,8 @@ int CNAME (BLASLONG m, BLASLONG n, BLASLONG k, FLOAT alpha, IFLOAT * A, IFLOAT * STORE_4X(0, 0, 1); STORE_4X(1, 0, 1); STORE_4X(0, 0, 2); STORE_4X(1, 0, 2); ptr_c += 16 * 2; - } - if (m > 31) { ptr_a0 = ptr_a1; + ptr_a1 = ptr_a0 + 16 * k; } for (; m_count > 15; m_count -= 16) { ptr_b0 = ptr_b00;