Skip to content

Commit 1d3e684

Browse files
committed
cpu: matmul: fix select postop failures for scalar broadcasting
1 parent 2cb38a2 commit 1d3e684

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

src/cpu/x64/matmul/brgemm_matmul.cpp

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -240,8 +240,16 @@ status_t brgemm_matmul_t<isa>::pd_t::init(engine_t *engine) {
240240
| primitive_attr_t::skip_mask_t::fpmath_mode,
241241
dst_dt),
242242
VERBOSE_UNSUPPORTED_ATTR);
243-
VDISPATCH_MATMUL(attr()->post_ops_.check_sum_consistency(dst_dt, is_int8),
243+
const auto &po = attr()->post_ops_;
244+
245+
VDISPATCH_MATMUL(po.check_sum_consistency(dst_dt, is_int8),
244246
VERBOSE_UNSUPPORTED_POSTOP);
247+
248+
VDISPATCH_MATMUL(
249+
!binary_injector::any_binary_postop_rhs_with_ternary_scalar_bcast(
250+
po, dst_d),
251+
VERBOSE_UNSUPPORTED_POSTOP);
252+
245253
VDISPATCH_MATMUL(check_attr_scales(), VERBOSE_UNSUPPORTED_SCALES_CFG);
246254
VDISPATCH_MATMUL(check_attr_zero_points(), VERBOSE_UNSUPPORTED_ZP_CFG);
247255
VDISPATCH_MATMUL(check_bias(), VERBOSE_UNSUPPORTED_BIAS_CFG);

0 commit comments

Comments
 (0)