fix zero padding pass addition for transpose stream layers

Jonathan-Shoemaker · Jonathan-Shoemaker · commit 8d42dc158d87 · 2022-07-25T12:06:18.000-05:00
diff --git a/hls4ml/backends/vivado/passes/conv_same_pad.py b/hls4ml/backends/vivado/passes/conv_same_pad.py
@@ -66,7 +66,7 @@ def transform(self, model, node):
         in_width = node.get_attr('in_width')
         stride_width = node.get_attr('stride_width')
         trfilt_width = (node.get_attr('filt_width') + node.get_attr('stride_width') - 1) \
-             // node.get_attr('stride_width')
+            // node.get_attr('stride_width')
 
         add_right = (convtr_out_width + pad_left)//stride_width - (in_width-1)
 
diff --git a/hls4ml/backends/vivado/vivado_backend.py b/hls4ml/backends/vivado/vivado_backend.py
@@ -39,6 +39,8 @@ def _register_flows(self):
             'vivado:clone_output',
             'vivado:insert_zero_padding_before_conv1d',
             'vivado:insert_zero_padding_before_conv2d',
+            'vivado:insert_zero_padding_before_conv1dtranspose',
+            'vivado:insert_zero_padding_before_conv2dtranspose',
             'vivado:broadcast_stream',
         ]
         streaming_flow = register_flow('streaming', streaming_passes, requires=[init_flow], backend=self.name)

Original file line number	Diff line number	Diff line change
`@@ -39,6 +39,8 @@ def _register_flows(self):`
`39`	`39`	`'vivado:clone_output',`
`40`	`40`	`'vivado:insert_zero_padding_before_conv1d',`
`41`	`41`	`'vivado:insert_zero_padding_before_conv2d',`
	`42`	`+ 'vivado:insert_zero_padding_before_conv1dtranspose',`
	`43`	`+ 'vivado:insert_zero_padding_before_conv2dtranspose',`
`42`	`44`	`'vivado:broadcast_stream',`
`43`	`45`	`]`
`44`	`46`	`streaming_flow = register_flow('streaming', streaming_passes, requires=[init_flow], backend=self.name)`