From 878ac2d4b2d561b44a9c2dc19f6988a7da0a71a6 Mon Sep 17 00:00:00 2001 From: Alok Bakshi Date: Fri, 27 Sep 2019 08:28:59 -0700 Subject: [PATCH] src: cpu: gemm: u8 grouped convolution fixup --- src/cpu/gemm_x8s8s32x_convolution.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/cpu/gemm_x8s8s32x_convolution.cpp b/src/cpu/gemm_x8s8s32x_convolution.cpp index e204a76c9c3..47f5cecb412 100644 --- a/src/cpu/gemm_x8s8s32x_convolution.cpp +++ b/src/cpu/gemm_x8s8s32x_convolution.cpp @@ -612,7 +612,7 @@ execute_forward_thr(const int ithr, const int nthr, const src_data_t *src_base, const int K = jcp.ks * jcp.ic; const int N = h_step * w_step; const int LDA = M * jcp.ngroups; - const int LDB = jcp.im2col_sz ? N : K; + const int LDB = jcp.im2col_sz ? N : K * jcp.ngroups; const char *BT = jcp.im2col_sz ? "T" : "N"; const int8_t off_a = 0, off_b = 0; const int32_t off_c = 0;