Skip to content

Commit 28d7a5e

Browse files
echereshdzarukin
authored andcommitted
cpu: ref_deconv: extend bias limitation check
1 parent 40de4ed commit 28d7a5e

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

src/cpu/ref_deconvolution.hpp

+5-1
Original file line numberDiff line numberDiff line change
@@ -492,7 +492,11 @@ struct ref_deconvolution_bwd_weights_t : public primitive_t {
492492
// if it was requested.
493493
if (with_bias()) {
494494
const auto bia_type = invariant_wei_md(1)->data_type;
495-
VDISPATCH_DECONVOLUTION(utils::one_of(bia_type, f32, bf16, f16),
495+
VDISPATCH_DECONVOLUTION(utils::one_of(bia_type, f32, bf16, f16)
496+
&& (bia_type == dst_type
497+
|| (bia_type == f32
498+
&& utils::one_of(
499+
dst_type, bf16, f16))),
496500
VERBOSE_UNSUPPORTED_DT);
497501
}
498502
VDISPATCH_DECONVOLUTION(utils::one_of(desc()->alg_kind,

0 commit comments

Comments
 (0)