From 3130d358064b876d17900f6a528a42616a7fed16 Mon Sep 17 00:00:00 2001 From: Pier Fiedorowicz Date: Fri, 16 Feb 2024 12:04:29 -0800 Subject: [PATCH] Allow asymmetric padding --- legacy/include/distconv/dnn_backend/convolution.hpp | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/legacy/include/distconv/dnn_backend/convolution.hpp b/legacy/include/distconv/dnn_backend/convolution.hpp index 045123bf..abbff46f 100644 --- a/legacy/include/distconv/dnn_backend/convolution.hpp +++ b/legacy/include/distconv/dnn_backend/convolution.hpp @@ -170,13 +170,12 @@ class Convolution } } - auto p = pads[0]; + bool use_padding = false; for (int i = 0; i < m_num_spatial_dims; ++i) { - assert_eq(pads[i], p); + use_padding = use_padding || (pads[i] != 0); assert_always(pads[i] == stencil_dims[i] || pads[i] == 0); } - bool use_padding = p != 0; internal::get_halo_sizes(input, IntVector(filter.get_shape()),