Skip to content

Commit

Permalink
x64: brgemm convolution: fix postops kernel creation condition
Browse files Browse the repository at this point in the history
  • Loading branch information
ankalinin authored and vpirogov committed Jun 1, 2023
1 parent c8943f5 commit 461d55e
Showing 1 changed file with 7 additions and 4 deletions.
11 changes: 7 additions & 4 deletions src/cpu/x64/jit_brgemm_conv.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -654,10 +654,13 @@ status_t brgemm_convolution_fwd_t<isa, use_inversion>::init(engine_t *engine) {
for_(int i_N = N_begin; i_N < N_end; i_N++)
for (int i_M = M_begin; i_M < M_end; i_M++) {
// init "init" and "po" kernels for cases then we never call brgemm kernels
// e.g. for d/h padded areas
const bool filter_in_padding = jcp.f_pad > EXT_KD
|| jcp.back_pad > EXT_KD || jcp.t_pad > EXT_KH
|| jcp.b_pad > EXT_KH;
// e.g. for d/h padded and dilated filter areas
const bool filter_in_padding = jcp.f_pad >= EXT_KD
|| jcp.back_pad >= EXT_KD || jcp.t_pad >= EXT_KH
|| jcp.b_pad >= EXT_KH;
// note: overly simplistic condition. Ideally, the condition would
// only detect cases where there is strictly no overlap between the
// input and filter.
const bool dilate_no_overlap
= jcp.dilate_d >= jcp.id || jcp.dilate_h >= jcp.ih;
if (IMPLICATION(jcp.exec_type == exec_trans,
Expand Down

0 comments on commit 461d55e

Please sign in to comment.