Raise an exception if the time budget is exceeded before the actual tuning starts

Author: fjwillemsen

kernel_tuner/interface.py

@@ -682,9 +682,16 @@ def tune_kernel(
     if verbose:
         print(f"Searchspace has {searchspace.size} configurations after restrictions.")
 
-    # call the strategy to execute the tuning process
-    tuning_options["start_time"] = perf_counter()
+    # register the times and raise an exception if the budget is exceeded
     tuning_options["startup_time"] = perf_counter() - start_overhead_time
+    if tuning_options["startup_time"] > tuning_options["time_limit"]:
+        raise RuntimeError(
+            f"The startup time of the tuning process ({tuning_options["startup_time"]} seconds) has exceeded the time limit ({tuning_options["time_limit"]} seconds). "
+            "Please increase the time limit or decrease the size of the search space."
+        )
+    tuning_options["start_time"] = perf_counter()
+
+    # call the strategy to execute the tuning process
     results = strategy.tune(searchspace, runner, tuning_options)
     env = runner.get_environment(tuning_options)
 

kernel_tuner/util.py

@@ -195,8 +195,6 @@ def check_stop_criterion(to):
     if "max_fevals" in to and len(to.unique_results) >= to.max_fevals:
         raise StopCriterionReached("max_fevals reached")
     if "time_limit" in to and (((time.perf_counter() - to.start_time) + (to.simulated_time * 1e-3) + to.startup_time) > to.time_limit):
-        if to.startup_time > to.time_limit:
-            raise RuntimeError(f"Time budget of {to.time_limit} seconds was already exceeded during startup ({to.startup_time} seconds).")
         raise StopCriterionReached("time limit exceeded")