From d5299f144f0dc8cdcca9195d8ffbbd5a77cff77e Mon Sep 17 00:00:00 2001
From: Arjen Roodselaar <arjen@oxide.computer>
Date: Mon, 19 Dec 2022 22:58:52 -0800
Subject: Add --no-placer-timeout flag to override timeout during refinement

---
 common/kernel/command.cc    |  4 ++++
 common/place/placer_heap.cc | 12 +++++++-----
 common/place/placer_heap.h  |  1 +
 3 files changed, 12 insertions(+), 5 deletions(-)

diff --git a/common/kernel/command.cc b/common/kernel/command.cc
index c548509f..22000299 100644
--- a/common/kernel/command.cc
+++ b/common/kernel/command.cc
@@ -189,6 +189,7 @@ po::options_description CommandHandler::getGeneralOptions()
     general.add_options()("placer-heap-critexp", po::value<int>(),
                           "placer heap criticality exponent (int, default: 2)");
     general.add_options()("placer-heap-timingweight", po::value<int>(), "placer heap timing weight (int, default: 10)");
+    general.add_options()("no-placer-timeout", "allow the placer to attempt placement without timeout");
 
 #if !defined(__wasm)
     general.add_options()("parallel-refine", "use new experimental parallelised engine for placement refinement");
@@ -328,6 +329,9 @@ void CommandHandler::setupContext(Context *ctx)
     if (vm.count("placer-heap-timingweight"))
         ctx->settings[ctx->id("placerHeap/timingWeight")] = std::to_string(vm["placer-heap-timingweight"].as<int>());
 
+    if (vm.count("no-placer-timeout"))
+        ctx->settings[ctx->id("placerHeap/noTimeout")] = true;
+
     if (vm.count("parallel-refine"))
         ctx->settings[ctx->id("placerHeap/parallelRefine")] = true;
 
diff --git a/common/place/placer_heap.cc b/common/place/placer_heap.cc
index 4ee9b909..69edd325 100644
--- a/common/place/placer_heap.cc
+++ b/common/place/placer_heap.cc
@@ -828,6 +828,11 @@ class HeAPPlacer
     // Strict placement legalisation, performed after the initial HeAP spreading
     void legalise_placement_strict(bool require_validity = false)
     {
+        int placement_timeout = cfg.no_placement_timeout ? 0 :
+            // Set a conservative timeout. This is a rather large number and could probably be
+            // shaved down, but for now it will keep the process from running indefinite.
+            std::max(10000, (int(ctx->cells.size()) * int(ctx->cells.size()) / 8) + 1);
+
         auto startt = std::chrono::high_resolution_clock::now();
 
         // Unbind all cells placed in this solution
@@ -879,11 +884,7 @@ class HeAPPlacer
             }
 
             while (!placed) {
-                // Set a conservative timeout. This is a rather large number and could probably be
-                // shaved down, but for now it will keep the process from running indefinite.
-                int timeout_limit = (int(ctx->cells.size()) * int(ctx->cells.size()) / 8) + 1;
-
-                if (total_iters_for_cell > std::max(10000, timeout_limit))
+                if (placement_timeout > 0 && total_iters_for_cell > placement_timeout)
                     log_error("Unable to find legal placement for cell '%s' after %d attempts, check constraints and utilisation.\n",
                               ctx->nameOf(ci), total_iters_for_cell);
 
@@ -1818,6 +1819,7 @@ PlacerHeapCfg::PlacerHeapCfg(Context *ctx)
     timing_driven = ctx->setting<bool>("timing_driven");
     solverTolerance = 1e-5;
     placeAllAtOnce = false;
+    no_placement_timeout = ctx->setting<bool>("placerHeap/noTimeout", false);
 
     hpwl_scale_x = 1;
     hpwl_scale_y = 1;
diff --git a/common/place/placer_heap.h b/common/place/placer_heap.h
index 9c62869e..680e1dc7 100644
--- a/common/place/placer_heap.h
+++ b/common/place/placer_heap.h
@@ -42,6 +42,7 @@ struct PlacerHeapCfg
     float solverTolerance;
     bool placeAllAtOnce;
     bool parallelRefine;
+    bool no_placement_timeout;
 
     int hpwl_scale_x, hpwl_scale_y;
     int spread_scale_x, spread_scale_y;
-- 
cgit v1.2.3