More connection rules for ikj crddrop support

weiya711 · Nov 22, 2023 · ef8043b · ef8043b
1 parent cc36ddf
commit ef8043b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/sam/onyx/hw_nodes/read_scanner_node.py b/sam/onyx/hw_nodes/read_scanner_node.py
@@ -169,6 +169,9 @@ def connect(self, other, edge, kwargs=None):
 
             if 'use_alt_out_port' in edge_attr:
                 out_conn = 'block_rd_out'
+            elif ('vector_reduce_mode' in edge_attr):
+                if (edge_attr['vector_reduce_mode']):
+                    out_conn = 'pos_out'
             else:
                 out_conn = 'coord_out'
 

diff --git a/sam/onyx/parse_dot.py b/sam/onyx/parse_dot.py
@@ -356,22 +356,22 @@ def rewrite_VectorReducer(self):
 
             print(crd_edge_attr)
             print(val_edge_attr)
-            del crd_edge_attr['comment']
 
             # crd_rd_scan_to_glb = pydot.Edge(src=crd_rd_scanner, dst=dst_crd, **crd_edge_attr, use_alt_out_port="1")
             # val_rd_scan_to_glb = pydot.Edge(src=vals_rd_scanner, dst=dst_vals, **val_edge_attr, use_alt_out_port="1")
 
+            # CRDDROP SUPPORT: TOOK OUT COMMENT ATTRIBUTE FROM BOTH OF THESE 
             crd_rd_scan_to_ds = pydot.Edge(
                 src=crd_rd_scanner,
                 dst=dst_crd,
                 **crd_edge_attr,
-                comment="final-crd",
+                #comment="final-crd",
                 vector_reduce_mode=True)
             val_rd_scan_to_ds = pydot.Edge(
                 src=vals_rd_scanner,
                 dst=dst_vals,
                 **val_edge_attr,
-                comment="final-val",
+                #comment="final-val",
                 vector_reduce_mode=True)
 
             # self.graph.add_edge(input_to_rsg_edge)