Merge pull request verilog-to-routing#3113 from verilog-to-routing/atom_pin_to_rr_node_id

[VPR][Utils] Add helper function to get RRNodeId from AtomPinId

Author: AlexandreSinger

libs/libarchfpga/src/physical_types.h

@@ -709,6 +709,7 @@ struct t_physical_tile_type {
 
     std::unordered_map<int, std::unordered_map<t_logical_block_type_ptr, t_pb_graph_pin*>> on_tile_pin_num_to_pb_pin; // [root_pin_physical_num][logical_block] -> t_pb_graph_pin*
     std::unordered_map<int, t_pb_graph_pin*> pin_num_to_pb_pin;                                                       // [intra_tile_pin_physical_num] -> t_pb_graph_pin
+    std::unordered_map<const t_pb_graph_pin*, int> pb_pin_to_pin_num;                                                 // [t_pb_graph_pin*] -> intra_tile_pin_physical_num
 
     std::vector<t_fc_specification> fc_specs;
 

vpr/src/base/SetupVPR.cpp

@@ -899,9 +899,11 @@ static void add_logical_pin_to_physical_tile(int physical_pin_offset,
                                              t_logical_block_type_ptr logical_block_ptr,
                                              t_physical_tile_type* physical_type) {
     for (auto logical_pin_pair : logical_block_ptr->pin_logical_num_to_pb_pin_mapping) {
-        auto pin_logical_num = logical_pin_pair.first;
-        auto pb_pin = logical_pin_pair.second;
-        physical_type->pin_num_to_pb_pin.insert(std::make_pair(pin_logical_num + physical_pin_offset, pb_pin));
+        int pin_logical_num = logical_pin_pair.first;
+        t_pb_graph_pin* pb_pin = logical_pin_pair.second;
+        int pin_physical_num = pin_logical_num + physical_pin_offset;
+        physical_type->pin_num_to_pb_pin.insert({pin_physical_num, pb_pin});
+        physical_type->pb_pin_to_pin_num.insert({pb_pin, pin_physical_num});
     }
 }
 

vpr/src/util/vpr_utils.cpp

@@ -1720,6 +1720,41 @@ RRNodeId get_class_rr_node_id(const RRSpatialLookup& rr_spatial_lookup,
     return rr_spatial_lookup.find_node(layer, i, j, node_type, class_physical_num);
 }
 
+RRNodeId get_atom_pin_rr_node_id(AtomPinId atom_pin_id) {
+    auto& atom_nlist = g_vpr_ctx.atom().netlist();
+    auto& atom_lookup = g_vpr_ctx.atom().lookup();
+    auto& place_ctx = g_vpr_ctx.placement();
+    auto& device_ctx = g_vpr_ctx.device();
+
+    /*
+     * To get the RRNodeId for an atom pin, we need to:
+     * 1. Find the atom block that the pin belongs to
+     * 2. Find the cluster block that the atom block is a part of
+     * 3. Find the physical tile that the cluster block is located on
+     * 4. Find the physical pin number of the atom pin (corresponds to ptc number of the RR node)
+     * 5. Call get_pin_rr_node_id to get the RRNodeId for the pin
+     */
+
+    AtomBlockId atom_blk_id = atom_nlist.pin_block(atom_pin_id);
+    ClusterBlockId clb_blk_id = atom_lookup.atom_clb(atom_blk_id);
+
+    t_pl_loc clb_blk_loc = place_ctx.block_locs()[clb_blk_id].loc;
+
+    t_physical_tile_type_ptr physical_tile = device_ctx.grid.get_physical_type({clb_blk_loc.x, clb_blk_loc.y, clb_blk_loc.layer});
+
+    const t_pb_graph_pin* atom_pb_pin = atom_lookup.atom_pin_pb_graph_pin(atom_pin_id);
+    int pin_physical_num = physical_tile->pb_pin_to_pin_num.at(atom_pb_pin);
+
+    RRNodeId rr_node_id = get_pin_rr_node_id(device_ctx.rr_graph.node_lookup(),
+                                             physical_tile,
+                                             clb_blk_loc.layer,
+                                             clb_blk_loc.x,
+                                             clb_blk_loc.y,
+                                             pin_physical_num);
+
+    return rr_node_id;
+}
+
 bool node_in_same_physical_tile(RRNodeId node_first, RRNodeId node_second) {
     const auto& device_ctx = g_vpr_ctx.device();
     const auto& rr_graph = device_ctx.rr_graph;

vpr/src/util/vpr_utils.h

@@ -248,6 +248,14 @@ RRNodeId get_pin_rr_node_id(const RRSpatialLookup& rr_spatial_lookup,
                             const int root_j,
                             int pin_physical_num);
 
+/**
+ * @brief Returns the RR node ID for the given atom pin ID.
+ * **Warning**: This function should be called only if flat-router is enabled,
+ * since, otherwise, the routing resources inside clusters are not added to the RR graph.
+ * @param atom_pin_id The atom pin ID.
+ */
+RRNodeId get_atom_pin_rr_node_id(AtomPinId atom_pin_id);
+
 RRNodeId get_class_rr_node_id(const RRSpatialLookup& rr_spatial_lookup,
                               t_physical_tile_type_ptr physical_tile,
                               const int layer,