diff --git a/src/finn/transformation/fpgadataflow/insert_dwc.py b/src/finn/transformation/fpgadataflow/insert_dwc.py index f86c62a9a6..0d8a0b001f 100644 --- a/src/finn/transformation/fpgadataflow/insert_dwc.py +++ b/src/finn/transformation/fpgadataflow/insert_dwc.py @@ -128,8 +128,11 @@ def apply(self, model): # determine dtype for dwc dtype = n0.get_output_datatype() n1_dtype = n1.get_input_datatype() - assert dtype == n1_dtype, f"Neighboring node datatypes are Incompatible ({dtype}) != ({n1_dtype})" - + assert dtype == n1_dtype, ( + "Neighboring node datatypes are Incompatible" + + f" ({dtype}) != ({n1_dtype})" + ) + # determine shapes for dwc # generalized version allows them to differ # and will either pad or crop depending @@ -145,10 +148,6 @@ def apply(self, model): ) graph.value_info.append(dwc_output_tensor) - print(f"inserting DWC_{style}, in_shape={in_shape},out_shape={out_shape},inWidth={dwc_in_width}, outWidth={dwc_out_width}, dtype={str(dtype.name)}") - #if str(dtype.name) == "UINT32": - # assert True == False - dwc_node = oh.make_node( node_optype, [output_name],