Allow 'parallelization_factor' propagation from dense layer

calad0i · calad0i · commit 4a356236dac8 · 2024-07-21T23:24:18.000-07:00
diff --git a/hls4ml/model/optimizer/passes/multi_dense.py b/hls4ml/model/optimizer/passes/multi_dense.py
@@ -27,6 +27,9 @@ def transform(self, model, node):
             'bias_data': node.get_attr('bias_data'),
         }
 
+        if (pf := node.get_attr('parallelization_factor', None)) is not None:
+            pointwise_attrs['parallelization_factor'] = pf
+
         if dim == 1:
             pointwise_attrs.update(
                 {

Original file line number	Diff line number	Diff line change
`@@ -27,6 +27,9 @@ def transform(self, model, node):`
`27`	`27`	`'bias_data': node.get_attr('bias_data'),`
`28`	`28`	`}`
`29`	`29`
	`30`	`+ if (pf := node.get_attr('parallelization_factor', None)) is not None:`
	`31`	`+ pointwise_attrs['parallelization_factor'] = pf`
	`32`	`+`
`30`	`33`	`if dim == 1:`
`31`	`34`	`pointwise_attrs.update(`
`32`	`35`	`{`