pass NoC link bandwidth utilization to graphics

Author: soheilshahrouz

vpr/src/draw/draw_global.cpp

@@ -33,6 +33,8 @@ static t_draw_coords draw_coords;
  */
 static std::optional<std::reference_wrapper<const BlkLocRegistry>> blk_loc_registry_ref;
 
+static std::optional<std::reference_wrapper<const vtr::vector<NocLinkId, double>>> noc_link_bandwidth_usages_ref;
+
 /*********************** Accessor Subroutines Definition ********************/
 
 /* This accessor function returns pointer to the global variable
@@ -55,4 +57,12 @@ const BlkLocRegistry& get_graphics_blk_loc_registry_ref() {
     return blk_loc_registry_ref->get();
 }
 
+void set_noc_link_bandwidth_usages_ref(const vtr::vector<NocLinkId, double>& noc_link_bandwidth_usages) {
+    noc_link_bandwidth_usages_ref = noc_link_bandwidth_usages;
+}
+
+const vtr::vector<NocLinkId, double>& get_noc_link_bandwidth_usages_ref() {
+    return noc_link_bandwidth_usages_ref->get();
+}
+
 #endif // NO_GRAPHICS

vpr/src/draw/draw_global.h

@@ -45,6 +45,10 @@ void set_graphics_blk_loc_registry_ref(const BlkLocRegistry& blk_loc_registry);
  */
 const BlkLocRegistry& get_graphics_blk_loc_registry_ref();
 
+void set_noc_link_bandwidth_usages_ref(const vtr::vector<NocLinkId, double>& noc_link_bandwidth_usages);
+
+const vtr::vector<NocLinkId, double>& get_noc_link_bandwidth_usages_ref();
+
 #endif // NO_GRAPHICS
 
 #endif

vpr/src/draw/draw_noc.cpp

@@ -75,6 +75,7 @@ void draw_noc(ezgl::renderer* g) {
 void draw_noc_usage(vtr::vector<NocLinkId, ezgl::color>& noc_link_colors) {
     t_draw_state* draw_state = get_draw_state_vars();
     auto& noc_ctx = g_vpr_ctx.noc();
+    const auto& noc_link_bandwidth_usages = get_noc_link_bandwidth_usages_ref();
 
     // check to see if a color map was already created previously
     if (draw_state->noc_usage_color_map == nullptr) {
@@ -90,15 +91,15 @@ void draw_noc_usage(vtr::vector<NocLinkId, ezgl::color>& noc_link_colors) {
     // represents the color to draw each noc link
     ezgl::color current_noc_link_color;
 
-    for (const auto& noc_link : noc_ctx.noc_model.get_noc_links()) {
-        NocLinkId link_id = noc_link.get_link_id();
+    for (const auto& [link_id, bandwidth_usage] : noc_link_bandwidth_usages.pairs()) {
 
         // only update the color of the link if it wasn't updated previously
         if (noc_link_colors[link_id] == ezgl::BLACK) {
             // if we are here then the link was not updated previously, so assign the color here
 
+            double link_bandwidth = noc_ctx.noc_model.get_single_noc_link(link_id).get_bandwidth();
             //get the current link bandwidth usage (ratio calculation)
-            double link_bandwidth_usage_ratio = 0.0;//(noc_link.get_bandwidth_usage()) / noc_link.get_bandwidth();
+            double link_bandwidth_usage_ratio = bandwidth_usage / link_bandwidth;
 
             // check if the link is being overused and if it is then cap it at 1.0
             if (link_bandwidth_usage_ratio > 1.0) {

vpr/src/place/noc_place_utils.cpp

@@ -800,6 +800,10 @@ double NocCostHandler::get_link_used_bandwidth(NocLinkId link_id) const {
     return link_bandwidth_usages[link_id];
 }
 
+const vtr::vector<NocLinkId, double>& NocCostHandler::get_link_bandwidth_usages() const {
+    return link_bandwidth_usages;
+}
+
 std::vector<NocLink> NocCostHandler::get_top_n_congested_links(int n) {
     // get NoC links
     vtr::vector<NocLinkId, NocLink> noc_links = g_vpr_ctx.noc().noc_model.get_noc_links();

vpr/src/place/noc_place_utils.h

@@ -286,6 +286,8 @@ class NocCostHandler {
      */
     double get_link_used_bandwidth(NocLinkId link_id) const;
 
+    const vtr::vector<NocLinkId, double>& get_link_bandwidth_usages() const;
+
     /**
      * @brief Determines the congestion cost a NoC link. The cost
      * is calculating by measuring how much the current bandwidth
@@ -510,9 +512,6 @@ class NocCostHandler {
 
     ///Represents the bandwidth of the data being transmitted on the link. Units in bits-per-second(bps)
     vtr::vector<NocLinkId, double> link_bandwidth_usages;
-
-
-    friend void test_revert_noc_traffic_flow_routes();
 };
 
 /**

vpr/src/place/place.cpp

@@ -439,6 +439,10 @@ void try_place(const Netlist<>& net_list,
     NetCostHandler net_cost_handler = alloc_and_load_placement_structs(placer_opts, noc_opts, directs,
                                                                        num_directs, placer_state, noc_cost_handler);
 
+    if (noc_cost_handler.has_value()) {
+        set_noc_link_bandwidth_usages_ref(noc_cost_handler->get_link_bandwidth_usages());
+    }
+
     ManualMoveGenerator manual_move_generator(placer_state);
 
     vtr::ScopedStartFinishTimer timer("Placement");
@@ -769,8 +773,8 @@ void try_place(const Netlist<>& net_list,
                                  *current_move_generator, manual_move_generator,
                                  blocks_affected, timing_info.get(),
                                  placer_opts.place_algorithm, move_type_stat,
-                                 timing_bb_factor,
-                                 swap_stats, placer_state, net_cost_handler, noc_cost_handler);
+                                 timing_bb_factor, swap_stats, placer_state,
+                                 net_cost_handler, noc_cost_handler);
 
 
             //move the update used move_generator to its original variable
@@ -837,8 +841,8 @@ void try_place(const Netlist<>& net_list,
                              *current_move_generator, manual_move_generator,
                              blocks_affected, timing_info.get(),
                              placer_opts.place_quench_algorithm, move_type_stat,
-                             timing_bb_factor,
-                             swap_stats, placer_state, net_cost_handler, noc_cost_handler);
+                             timing_bb_factor, swap_stats, placer_state,
+                             net_cost_handler, noc_cost_handler);
 
 
         //move the update used move_generator to its original variable