add parallel_segment_index and chan_type member vars to t_bottleneck_link

Author: soheilshahrouz

vpr/src/route/rr_graph_generation/build_scatter_gathers.cpp

@@ -7,6 +7,7 @@
 #include "scatter_gather_types.h"
 #include "rr_types.h"
 #include "globals.h"
+#include "rr_graph_uxsdcxx_interface.h"
 #include "vtr_assert.h"
 #include "vtr_random.h"
 
@@ -160,7 +161,9 @@ static std::vector<t_sg_candidate> find_candidate_wires(const std::vector<t_chan
 
 std::vector<t_bottleneck_link> alloc_and_load_scatter_gather_connections(const std::vector<t_scatter_gather_pattern>& scatter_gather_patterns,
                                                                          const std::vector<bool>& inter_cluster_rr,
-                                                                         const std::vector<t_segment_inf>& segment_inf,
+                                                                         const std::vector<t_segment_inf>& segment_inf_x,
+                                                                         const std::vector<t_segment_inf>& segment_inf_y,
+                                                                         const std::vector<t_segment_inf>& segment_inf_z,
                                                                          const t_chan_details& chan_details_x,
                                                                          const t_chan_details& chan_details_y,
                                                                          const t_chan_width& nodes_per_chan,
@@ -201,6 +204,19 @@ std::vector<t_bottleneck_link> alloc_and_load_scatter_gather_connections(const s
                 VTR_ASSERT(sg_link_it != sg_pattern.sg_links.end());
                 const t_sg_link& sg_link = *sg_link_it;
 
+                VTR_ASSERT(vtr::exactly_k_conditions(1, sg_link.x_offset != 0, sg_link.y_offset != 0, sg_link.z_offset != 0));
+                t_physical_tile_loc scatter_loc;
+                scatter_loc.x = gather_loc.x + sg_link.x_offset;
+                scatter_loc.y = gather_loc.y + sg_link.y_offset;
+                scatter_loc.layer_num = gather_loc.layer_num + sg_link.z_offset;
+
+                const std::vector<t_segment_inf> segment_inf& = (sg_link.x_offset != 0) ? segment_inf_x :
+                                                                (sg_link.y_offset != 0) ? segment_inf_y : segment_inf_z;
+
+                const e_rr_type chan_type = (sg_link.x_offset != 0) ? e_rr_type::CHANX :
+                                            (sg_link.y_offset != 0) ? e_rr_type::CHANY : e_rr_type::CHANZ;
+
+
                 auto seg_it = std::ranges::find_if(segment_inf,
                                                    [&](const t_segment_inf& seg) noexcept {
                                                        return seg.name == sg_link.seg_type;
@@ -209,10 +225,6 @@ std::vector<t_bottleneck_link> alloc_and_load_scatter_gather_connections(const s
                 VTR_ASSERT(seg_it != segment_inf.end());
                 const t_segment_inf& wire_segment = *seg_it;
 
-                t_physical_tile_loc scatter_loc;
-                scatter_loc.x = gather_loc.x + sg_link.x_offset;
-                scatter_loc.y = gather_loc.y + sg_link.y_offset;
-                scatter_loc.layer_num = gather_loc.layer_num + sg_link.z_offset;
 
                 index_to_correct_sg_channels(sg_pattern.gather_pattern, gather_loc, chan_details_x, chan_details_y, gather_channels);
                 index_to_correct_sg_channels(sg_pattern.scatter_pattern, scatter_loc, chan_details_x, chan_details_y, scatter_channels);
@@ -317,6 +329,9 @@ std::vector<t_bottleneck_link> alloc_and_load_scatter_gather_connections(const s
                         } else {
                             bottleneck_link.arch_wire_switch = wire_segment.arch_wire_switch;
                         }
+
+                        bottleneck_link.chan_type = chan_type;
+                        bottleneck_link.parallel_segment_index = std::distance(segment_inf.begin(), seg_it);
                         bottleneck_links.push_back(std::move(bottleneck_link));
                     }
                 }

vpr/src/route/rr_graph_generation/build_scatter_gathers.h

@@ -7,7 +7,7 @@
 
 #include <vector>
 
-// forward declartion
+// forward declaration
 namespace vtr {
 class RngContainer;
 }
@@ -32,6 +32,8 @@ struct t_bottleneck_link {
     t_physical_tile_loc gather_loc;                         ///< Source switchblock location.
     t_physical_tile_loc scatter_loc;                        ///< Destination switchblock location.
     int arch_wire_switch;                                   ///< The switch (mux) used to drive the bottleneck wire.
+    int parallel_segment_index;
+    e_rr_type chan_type;
     std::vector<t_sg_candidate> gather_fanin_connections;   ///< Wires driving the bottleneck link at  `gather_loc`
     std::vector<t_sg_candidate> scatter_fanout_connections; ///< Wires driven by the bottleneck link at `scatter_loc`
 };
@@ -46,7 +48,7 @@ struct t_bottleneck_link {
  *
  * @param scatter_gather_patterns List of scatter/gather connection patterns.
  * @param inter_cluster_rr Flags indicating whether each layer has inter-cluster routing resources.
- * @param segment_inf Wire segment type information.
+ * @param segment_inf_x, segment_inf_y, segment_inf_z Wire segment type information along each axis.
  * @param chan_details_x Channel details for horizontal routing channels.
  * @param chan_details_y Channel details for vertical routing channels.
  * @param nodes_per_chan Channel width data.
@@ -56,7 +58,9 @@ struct t_bottleneck_link {
  */
 std::vector<t_bottleneck_link> alloc_and_load_scatter_gather_connections(const std::vector<t_scatter_gather_pattern>& scatter_gather_patterns,
                                                                          const std::vector<bool>& inter_cluster_rr,
-                                                                         const std::vector<t_segment_inf>& segment_inf,
+                                                                         const std::vector<t_segment_inf>& segment_inf_x,
+                                                                         const std::vector<t_segment_inf>& segment_inf_y,
+                                                                         const std::vector<t_segment_inf>& segment_inf_z,
                                                                          const t_chan_details& chan_details_x,
                                                                          const t_chan_details& chan_details_y,
                                                                          const t_chan_width& nodes_per_chan,

vpr/src/route/rr_graph_generation/rr_graph.cpp

@@ -913,7 +913,7 @@ static void build_rr_graph(e_graph_type graph_type,
     vtr::NdMatrix<std::vector<t_bottleneck_link>, 2> interdie_3d_links;
     const std::vector<t_bottleneck_link> bottleneck_links = alloc_and_load_scatter_gather_connections(scatter_gather_patterns,
                                                                                                       inter_cluster_prog_rr,
-                                                                                                      segment_inf,
+                                                                                                      segment_inf_x, segment_inf_y, segment_inf_z,
                                                                                                       chan_details_x, chan_details_y,
                                                                                                       nodes_per_chan,
                                                                                                       switchpoint_rng,