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

Author: fjwillemsen

kernel_tuner/interface.py

@@ -683,12 +683,13 @@ def tune_kernel(
         print(f"Searchspace has {searchspace.size} configurations after restrictions.")
 
     # 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."
-        )
+    if "time_limit" in tuning_options:
+        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