Skip to content

Commit

Permalink
further print cleanup
Browse files Browse the repository at this point in the history
  • Loading branch information
lstasytis committed Nov 26, 2024
1 parent 38f897c commit aad5685
Showing 1 changed file with 5 additions and 6 deletions.
11 changes: 5 additions & 6 deletions src/finn/transformation/fpgadataflow/insert_dwc.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -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],
Expand Down

0 comments on commit aad5685

Please sign in to comment.