diff --git a/src/finn/transformation/fpgadataflow/derive_characteristic.py b/src/finn/transformation/fpgadataflow/derive_characteristic.py index f857cdb5ef7328bec34b798010d3fbb167a61208..65146206641f4e701f44c701d843028647f0d260 100644 --- a/src/finn/transformation/fpgadataflow/derive_characteristic.py +++ b/src/finn/transformation/fpgadataflow/derive_characteristic.py @@ -188,7 +188,7 @@ class DeriveCharacteristic(NodeLocalTransformation): (model, run_again) = super().apply(model) # apply manual fix for DuplicateStreams and AddStreams for # simple residual reconvergent paths with bypass - addstrm_nodes = model.get_nodes_by_op_type("AddStreams") + addstrm_nodes = model.get_nodes_by_op_type("AddStreams_Batch") for addstrm_node in addstrm_nodes: # we currently only support the case where one branch is # a bypass @@ -197,8 +197,8 @@ class DeriveCharacteristic(NodeLocalTransformation): if (b0 is None) or (b1 is None): warnings.warn("Found unsupported AddStreams, skipping") return (model, run_again) - b0_is_bypass = b0.op_type == "DuplicateStreams" - b1_is_bypass = b1.op_type == "DuplicateStreams" + b0_is_bypass = b0.op_type == "DuplicateStreams_Batch" + b1_is_bypass = b1.op_type == "DuplicateStreams_Batch" if (not b0_is_bypass) and (not b1_is_bypass): warnings.warn("Found unsupported AddStreams, skipping") return (model, run_again)