Remove comments

Xilinx · Sep 12, 2023 · 23d5743 · 23d5743
1 parent 1a106ac
commit 23d5743
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 5 deletions.
diff --git a/src/brevitas/graph/equalize.py b/src/brevitas/graph/equalize.py
@@ -681,7 +681,6 @@ def cat_handler(graph_model: GraphModule, starting_node: Node, state: WalkRegion
     state.srcs.clear()
     state.sinks.clear()
     state.history.clear()
-    # state.srcs.add((starting_node.target, ()))
     state.srcs[starting_node.target] = _UNSUPPORTED_OP
     state.update_offset = True
     state.offset = 0
@@ -837,7 +836,8 @@ def _extract_regions(
                 module = get_module(graph_model, node.target)
                 weight = get_weight_source([module])
                 eq_indexes = EqualizationIndexes(0, weight[0].shape[0], 0)
-                state.srcs[node.target] = eq_indexes
+                full_source_name = node.target + '$' + str(eq_indexes)
+                state.srcs[full_source_name] = eq_indexes
             find_sinks(graph_model, node, state)
             if state.sinks and _UNSUPPORTED_OP not in state.sinks.keys(
             ) and _UNSUPPORTED_OP not in state.srcs.keys():
@@ -851,10 +851,8 @@ def _extract_regions(
                 else:
                     region = Region(srcs=sorted_srcs, sinks=sorted_sinks)
 
-                # new_region = (sorted_tuple_srcs, sorted_tuple_sinks)
                 if region not in regions:
                     regions.append(region)
-                    # regions_name.add(region_names)
     return regions
 
 

diff --git a/src/brevitas/graph/test_unet.py b/src/brevitas/graph/test_unet.py
@@ -44,7 +44,6 @@ def main(args):
 
     # Perform equalization
     model, regions = EqualizeGraph(iterations=20, return_regions=True).apply(model)
-    print(len(regions))
     # New output after equalization
     out = model(x)
     # for region in regions: