diff --git a/src/finn/custom_op/fpgadataflow/convolutioninputgenerator1d.py b/src/finn/custom_op/fpgadataflow/convolutioninputgenerator1d.py index 89b3fd45ea9b5231b4ddb3e595bf18b18d07ab60..6347c9e9e6923cff6c1d02d272030fbdb100604a 100644 --- a/src/finn/custom_op/fpgadataflow/convolutioninputgenerator1d.py +++ b/src/finn/custom_op/fpgadataflow/convolutioninputgenerator1d.py @@ -217,12 +217,13 @@ class ConvolutionInputGenerator1D(HLSCustomOp): dilation = self.get_nodeattr("Dilation") stride_h, stride_w = stride dilation_h, dilation_w = dilation + ram_style = self.get_nodeattr("ram_style") if self.get_nodeattr("SIMD") == self.get_nodeattr("IFMChannels"): if self.get_nodeattr("depthwise") == 0: if stride_h == 1 and stride_w == 1: if dilation_h == 1 and dilation_w == 1: - return True + return ram_style in ["auto", "distributed"] return False