Skip to content

Commit b8e6ca6

Browse files
committed
Fix inner product failure when original weight is f16 and oc is 1
1 parent 5baba71 commit b8e6ca6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/cpu/x64/jit_brgemm_inner_product_utils.cpp

+1-1
Original file line numberDiff line numberDiff line change
@@ -525,7 +525,7 @@ status_t jit_brgemm_ip_fwd_conf_t::init_conf(cpu_isa_t isa,
525525
jbgp.nb_ic = div_up(jbgp.ic, jbgp.ic_block);
526526

527527
// gemm-based inner product performs better when oc = 1
528-
if (is_f32_compute && jbgp.oc == 1) return status::unimplemented;
528+
if (is_f32_compute && jbgp.oc == 1 && jbgp.orig_wei_dt != f16) return status::unimplemented;
529529

530530
jbgp.oc_block = get_adjusted_oc_block();
531531
jbgp.nb_oc = div_up(jbgp.oc, jbgp.oc_block);

0 commit comments

Comments
 (0)