feat: support custom initial reroute queue.

Signed-off-by: zarkin404 <[email protected]>

Author: zarkin404

src/drt/include/triton_route/TritonRoute.h

@@ -137,11 +137,11 @@ class TritonRoute
   void setBreakIter(int breakIter);
   void setParallelWorkers(int parallelWorkers);
   void setSkipReroute(bool on = true);
-  void setRerouteNetsSortMode(int sortMode);
+  void setRerouteNetsSortMode(int mode);
   void setApiHost(const std::string& apiHost);
   void setApiTimeout(int timeout);
   void setNetOrderingTraining(bool on = true);
-  void setNetOrderingEvaluation(bool on = true);
+  void setNetOrderingEvaluation(int mode);
   void setDebugPaMarkers(bool on = true);
   void setDebugWorkerParams(int mazeEndIter,
                             int drcCost,

src/drt/src/TritonRoute.cpp

@@ -206,9 +206,9 @@ void TritonRoute::setNetOrderingTraining(bool on)
   debug_->netOrderingTraining = on;
 }
 
-void TritonRoute::setNetOrderingEvaluation(bool on)
+void TritonRoute::setNetOrderingEvaluation(int mode)
 {
-  debug_->netOrderingEvaluation = on;
+  debug_->netOrderingEvaluation = mode;
 }
 
 void TritonRoute::setDebugPaMarkers(bool on)

src/drt/src/TritonRoute.i

@@ -174,7 +174,7 @@ set_detailed_route_debug_cmd(const char* net_name,
                              const char* api_host,
                              int api_timeout,
                              bool net_ordering_training,
-                             bool net_ordering_evaluation)
+                             int net_ordering_evaluation)
 {
   auto* router = ord::OpenRoad::openRoad()->getTritonRoute();
   router->setDebugNetName(net_name);

src/drt/src/TritonRoute.tcl

@@ -224,10 +224,10 @@ proc detailed_route_debug { args } {
   sta::parse_key_args "detailed_route_debug" args \
       keys {-net -worker -iter -pin -dump_dir -custom_size -custom_offset \
             -break_iter -parallel_workers -reroute_nets_sort_mode \
-            -api_host -api_timeout} \
+            -api_host -api_timeout -net_ordering_evaluation} \
       flags {-dr -maze -pa -pa_markers -pa_edge -pa_commit -dump_dr -ta \
              -custom_strategies -skip_reroute \
-             -net_ordering_training -net_ordering_evaluation}
+             -net_ordering_training}
 
   sta::check_argc_eq0 "detailed_route_debug" $args
 
@@ -243,7 +243,6 @@ proc detailed_route_debug { args } {
   set skip_reroute [info exists flags(-skip_reroute)]
   set skip_sort_reroute_nets [info exists flags(-skip_sort_reroute_nets)]
   set net_ordering_training [info exists flags(-net_ordering_training)]
-  set net_ordering_evaluation [info exists flags(-net_ordering_evaluation)]
 
   if { [info exists keys(-net)] } {
     set net_name $keys(-net)
@@ -303,6 +302,12 @@ proc detailed_route_debug { args } {
     set parallel_workers 0
   }
 
+  if { [info exists keys(-net_ordering_evaluation)] } {
+    set net_ordering_evaluation $keys(-net_ordering_evaluation)
+  } else {
+    set net_ordering_evaluation 0
+  }
+
   if { [info exists keys(-reroute_nets_sort_mode)] } {
     set reroute_nets_sort_mode $keys(-reroute_nets_sort_mode)
   } else {

src/drt/src/dr/FlexDR_init.cpp

@@ -2531,6 +2531,7 @@ void FlexDRWorker::route_queue_init_queue(queue<RouteQueueEntry>& rerouteQueue)
 
   if (getRipupMode() == 0) {
     for (auto& marker : markers_) {
+      // checks 中的元素为的本轮不布的网络
       route_queue_update_from_marker(
           &marker, uniqueVictims, uniqueAggressors, checks, routes);
     }
@@ -2557,6 +2558,82 @@ void FlexDRWorker::route_queue_init_queue(queue<RouteQueueEntry>& rerouteQueue)
   } else {
     cout << "Error: unsupported ripup mode\n";
   }
+
+  vector<unsigned int> customOrder; // 记录算法端给出的排序
+  vector<RouteQueueEntry> customOrderRoutes;  // 记录即将替换掉初始排序的临时序列
+
+  // 当推断模式为「替换初始排序」时，使用算法排序好的序列
+  if (debugSettings_->apiHost != ""
+      && debugSettings_->netOrderingEvaluation == 2) {
+    std::string addr = "tcp://" + debugSettings_->apiHost;
+    utl::MQ mq(addr, debugSettings_->apiTimeout);
+
+    // 先将不需要布线的网络放到队列
+    for (auto& entry : checks) {
+      customOrderRoutes.push_back(entry);
+    }
+
+    // 原顺序为 | routes | checks ，现在为 | checks | routes (#pin < 2) | routes (#pin >= 2)
+
+    // 待布网络下标
+    vector<unsigned int> outerNetIdxRemaining;
+    // 构建外部 net 的索引
+    map<unsigned int, drNet*> outerNetMap;
+    int outerNetIdx = 0;
+    for (auto& entry : routes) {
+      frBlockObject* obj = entry.block;
+      bool doRoute = entry.doRoute;
+
+      if (doRoute) {
+        auto net = static_cast<drNet*>(obj);
+        if (net->getPins().size() > 1) {
+          outerNetIdxRemaining.push_back(outerNetIdx);
+          outerNetMap[outerNetIdx] = net;
+          outerNetIdx++;
+        } else {
+          customOrderRoutes.push_back(entry);
+        }
+      } else {
+        customOrderRoutes.push_back(entry);
+      }
+    }
+
+    // 仅处理有 1 个待布网络以上的情况
+    if (outerNetIdxRemaining.size() > 1) {
+      setOuterNetMap(outerNetMap);
+
+      openroad_api::net_ordering::Message msg;
+      openroad_api::net_ordering::Request* req = msg.mutable_request();
+
+      // 获取 gridGraph 的数据
+      gridGraph_.dump(req);
+
+      req->mutable_nets()->CopyFrom({outerNetIdxRemaining.begin(),
+                                     outerNetIdxRemaining.end()});
+
+      string reqStr = msg.SerializeAsString();
+      logger_->info(DRT, 991, "Requesting initial net order...");
+      string res = mq.request(reqStr);
+      msg = openroad_api::net_ordering::Message();
+      msg.ParseFromString(res);
+      auto netOrder = msg.response().net_list();
+
+      for (auto& netIdx : netOrder) {
+        customOrder.push_back(netIdx);
+      }
+    }
+  }
+
+  // 如果有排序结果，则替换为算法给出的排序
+  if (customOrder.size() > 0) {
+    for (auto& netIdx : customOrder) {
+      customOrderRoutes.push_back({outerNetMap_[netIdx], 0, true});
+    }
+
+    routes = customOrderRoutes;
+  }
+
+  // 依次将 route 和 check 中的元素压入队列
   route_queue_update_queue(checks, routes, rerouteQueue);
 }
 

src/drt/src/dr/FlexDR_maze.cpp

@@ -1797,7 +1797,10 @@ void FlexDRWorker::route_queue_main(queue<RouteQueueEntry>& rerouteQueue)
 {
   auto& workerRegionQuery = getWorkerRegionQuery();
 
-  if (debugSettings_->apiHost != "") {
+  // 训练模式，或者单步的推断模式
+  if (debugSettings_->apiHost != ""
+      && (debugSettings_->netOrderingTraining
+          || debugSettings_->netOrderingEvaluation == 1)) {
     std::string addr = "tcp://" + debugSettings_->apiHost;
     utl::MQ mq(addr, debugSettings_->apiTimeout);
 

src/drt/src/frBaseTypes.h

@@ -315,9 +315,9 @@ struct frDebugSettings
         parallelWorkers(0),
         skipReroute(false),
         rerouteNetsSortMode(0),
-        apiTimeout(0),
+        apiTimeout(30000),
         netOrderingTraining(false),
-        netOrderingEvaluation(false),
+        netOrderingEvaluation(0),
         paMarkers(false),
         paEdge(false),
         paCommit(false),
@@ -356,7 +356,7 @@ struct frDebugSettings
   std::string apiHost;
   int apiTimeout;
   bool netOrderingTraining;
-  bool netOrderingEvaluation;
+  int netOrderingEvaluation;
   bool paMarkers;
   bool paEdge;
   bool paCommit;