Skip to content

Commit dc6c0f3

Browse files
rjourslertprimak
authored andcommitted
tests: conv: Fix dispatch for fast ref bwd_d convolution dispatching
1 parent cbffe3c commit dc6c0f3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/benchdnn/conv/ref_conv.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -113,7 +113,7 @@ void compute_ref_direct_fwd(const prb_t *p, dnn_mem_t &src_m, dnn_mem_t &wei_m,
113113
void compute_ref_direct_bwd_d(const prb_t *p, dnn_mem_t &diff_src_m,
114114
dnn_mem_t &wei_m, dnn_mem_t &bia_m, dnn_mem_t &diff_dst_m) {
115115
enum { precompute_size = 16 };
116-
const bool fast = MAX2(p->kh, p->kw) <= precompute_size;
116+
const bool fast = MAX3(p->kd, p->kh, p->kw) <= precompute_size;
117117

118118
/* pre-computes arrays of oh(ow) and kh(kw) for traversing in kernel */
119119
auto precompute_ok

0 commit comments

Comments
 (0)