cpu: x64: brgemm_ip_fwd: fallback to gemm-based ip when oc = 1 for f32

This commit is contained in:
Bhavani Subramanian
2021-04-27 12:07:33 -10:00
committed by tprimak
parent 3555d4a76e
commit 714b1fd7f9
@@ -140,6 +140,10 @@ status_t init_ip_conf_fwd(
constexpr int amx_int8_row = 64;
jbgp.ic_block = (is_amx_int8) ? amx_int8_row : jbgp.simd_w;
// gemm-based inner product performs better when oc = 1
if (is_f32 && jbgp.oc == 1) return status::unimplemented;
if (jbgp.oc >= 64) {
jbgp.oc_block = 64;
} else if (jbgp.oc >= 32) {