frozenleaves/oneDNN
1
0
Fork 0
mirror of https://github.com/uxlfoundation/oneDNN.git synced 2025-10-20 10:03:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

cpu: x64: brgconv_bwd_strided: limit bias to f32 only for f32 config

Browse Source
This commit is contained in:
Dmitrii Zarukin
2025-01-09 15:53:01 -08:00
committed by Dmitry Zarukin
parent 9f3ff804c4
commit c563f8e36c
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/cpu/x64/jit_brgemm_conv_bwd_strided.cpp
View File

@ -91,7 +91,8 @@ status_t brgemm_convolution_bwd_strided_t<isa>::pd_t::init(engine_t *engine) {
| skip_mask_t::zero_points_runtime;
const bool is_f32_supported
= everyone_is(f32, diff_src_type, wei_type, diff_dst_type);
= everyone_is(f32, diff_src_type, wei_type, diff_dst_type)
&& IMPLICATION(with_bias(), bias_md_.data_type == f32);
const bool is_xf16_supported = one_of(wei_type, bf16, f16)
&& wei_type == diff_dst_type && one_of(diff_src_type, wei_type, f32)