fix cpu_budget default to be None and reduce timeout fail

Signed-off-by: Jack Luar <[email protected]>

Author: luarss

tools/AutoTuner/src/autotuner/distributed.py

@@ -66,14 +66,14 @@
 
 TEMPLATE = """
 Expected figures for this experiment.
-    Wall time: {runtime:.2f} hours
+    Wall time: {runtime:.5f} hours
     Number of Samples: 
-        Samples per minute: {num_samples_per_minute:.2f}
-        Design runtime of 10 min: {num_samples_10min:.2f}
-        Design runtime of 1h: {num_samples_1h:.2f}
+        Samples per minute: {num_samples_per_minute:.5f}
+        Design runtime of 10 min: {num_samples_10min:.5f}
+        Design runtime of 1h: {num_samples_1h:.5f}
     Number of iterations
-        Design runtime of 10 min: {num_iterations_10min:.2f}
-        Design runtime of 1h: {num_iterations_1h:.2f}
+        Design runtime of 10 min: {num_iterations_10min:.5f}
+        Design runtime of 1h: {num_iterations_1h:.5f}
 """
 
 
@@ -855,7 +855,7 @@ def parse_arguments():
         "--cpu_budget",
         type=int,
         metavar="<int>",
-        default=int(np.floor(cpu_count() / 2)),
+        default=None,
         help="CPU Hours",
     )
     parser.add_argument(
@@ -1051,6 +1051,8 @@ def sweep():
 
 if __name__ == "__main__":
     args = parse_arguments()
+    print(args)
+    exit()
 
     # Read config and original files before handling where to run in case we
     # need to upload the files.

tools/AutoTuner/test/smoke_test_timeout.py

@@ -18,14 +18,15 @@ def setUp(self):
         )
         self.experiment = f"smoke-test-timeout-{self.platform}"
 
-        # 0.01 hour translates to 36 seconds, which will definitely cause failure.
-        timeout_flags = ["--timeout 0.05", "--timeout_per_trial 0.05"]
+        # 0.001 hour translates to 3.6 seconds, which will definitely cause failure.
+        timeout_flags = ["--timeout 0.001", "--timeout_per_trial 0.001"]
         self.commands = [
             "python3 distributed.py"
             f" --design {self.design}"
             f" --platform {self.platform}"
             f" --experiment {self.experiment}"
             f" --config {self.config}"
+            f" --cpu_budget None"
             f" --yes"
             f" {flag}"
             f" tune --samples 1"