diff --git a/src/gpu/jit/conv/config_plan.cpp b/src/gpu/jit/conv/config_plan.cpp index d8ce7fb0372..7df7d7bd573 100644 --- a/src/gpu/jit/conv/config_plan.cpp +++ b/src/gpu/jit/conv/config_plan.cpp @@ -2080,7 +2080,7 @@ class plan_builder_t { vec_size, a_layout, b_layout, 1, -1)) { m_blk = vec_size; c_blk_layout = c_blk_layout.add_outer_block(1, vec_size); - } else if (cfg_.prb().g > 1 + } else if (cfg_.prb().g > 1 && cfg_.prb().g < 16 && (a_layout.blocks()[0].block * a_layout.blocks()[1].block) % vec_size