cpu: x64: brgemm_ip_fwd: fallback to gemm-based ip when oc = 1 for f32
This commit is contained in:
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) {
|
||||
|
||||
Reference in New Issue
Block a user