Skip to content

Commit b134a44

Browse files
[FORK][FIX][x64] enable ncx8c->ncx8c support for small input channel in jit sse41 conv
CVS-ticket https://jira.devtools.intel.com/browse/CVS-125793 openvino PR openvinotoolkit/openvino#23232 Signed-off-by: HU Yuan2 <yuan2.hu@intel.com>
1 parent 26633ae commit b134a44

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

src/cpu/x64/jit_sse41_conv_kernel_f32.cpp

+1-1
Original file line numberDiff line numberDiff line change
@@ -492,7 +492,7 @@ status_t jit_sse41_conv_fwd_kernel_f32::init_conf(jit_conv_conf_t &jcp,
492492
sum_requires_scale_one, sum_requires_zp_zero));
493493
if (!post_ops_ok_) return status::unimplemented;
494494

495-
const bool flat = one_of(jcp.ic, 1, 2, 3);
495+
const bool flat = one_of(jcp.ic, 1, 2, 3) && (jcp.src_tag != dat_tag_nCx8c);
496496
const bool mimo = !flat;
497497

498498
bool args_ok = true

src/cpu/x64/jit_sse41_convolution.hpp

+1-1
Original file line numberDiff line numberDiff line change
@@ -85,7 +85,7 @@ struct jit_sse41_convolution_fwd_t : public primitive_t {
8585
&& IMPLICATION(curr_dst_tag != dat_tag_nxc,
8686
dst_d.format_kind() == format_kind::any)
8787
&& utils::one_of(dat_tag_nxc, curr_src_tag, curr_dst_tag);
88-
const bool flat = utils::one_of(IC(), 1, 2, 3);
88+
const bool flat = utils::one_of(IC(), 1, 2, 3) && (curr_src_tag != dat_tag_nCx8c);
8989
auto src_tag = is_data_layout_nxc ? dat_tag_nxc
9090
: flat ? dat_tag_ncx
9191
: dat_tag_nCx8c;

0 commit comments

Comments
 (0)