Skip to content

Commit 9534621

Browse files
committed
cpu: gemm: check for nocopy on sequential mode
We should check for nocopy even for single threaded mode.
1 parent 5994d63 commit 9534621

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

src/cpu/gemm/gemm_driver.cpp

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1667,11 +1667,6 @@ static mkldnn_status_t gemm_threading_driver(
16671667
}
16681668
get_omp_thread_count<c_type>(arg->m, arg->n, arg->k, &nthr);
16691669

1670-
if (nthr == 1) {
1671-
return gemm_kernel_driver(arg->m, arg->n, arg->k, arg->a, arg->b,
1672-
arg->c, arg->co, arg);
1673-
}
1674-
16751670
if ((data_traits<a_type>::data_type == data_type::f32) &&
16761671
nocopy_checker(nthr, arg->transa, arg->transb, arg->m, arg->n,
16771672
arg->k, arg->lda, arg->ldb, arg->ldc))
@@ -1681,6 +1676,11 @@ static mkldnn_status_t gemm_threading_driver(
16811676
(float *) arg->b, arg->ldb,
16821677
arg->beta, (float *) arg->c, arg->ldc, NULL);
16831678

1679+
if (nthr == 1) {
1680+
return gemm_kernel_driver(
1681+
arg->m, arg->n, arg->k, arg->a, arg->b, arg->c, arg->co, arg);
1682+
}
1683+
16841684
mkldnn_status_t *results = (mkldnn_status_t *) malloc(
16851685
sizeof(*results) * nthr * CACHE_LINE_SIZE, PAGE_4K);
16861686

0 commit comments

Comments
 (0)