diff --git a/src/finn/transformation/fpgadataflow/convert_to_hls_layers.py b/src/finn/transformation/fpgadataflow/convert_to_hls_layers.py index 6cd555c52c17e86a7d0e0da7040322636c951f7f..2f83f23cb51248c6a6e6a9a1233f41435c3bf966 100644 --- a/src/finn/transformation/fpgadataflow/convert_to_hls_layers.py +++ b/src/finn/transformation/fpgadataflow/convert_to_hls_layers.py @@ -213,10 +213,11 @@ class InferConvInpGen(Transformation): 1D depthwise separable convolutions""" % n.name ) - if stride_h > 1 or stride_w > 1: + if (stride_h > 1 or stride_w > 1) and (not depthwise): assert ( stride_h < k_h and stride_w < k_w - ), """%s: Stride value must be smaller than kernel dim""" + ), """%s: Stride value must be smaller than kernel dim + for non-depthwise (dense) convolutions""" ConvInpGen_node = helper.make_node( "ConvolutionInputGenerator1D", [ConvInpGen_input],