consistent file names and fix smoke_test_sweep

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

Author: luarss

tools/AutoTuner/src/autotuner/distributed.py

@@ -404,19 +404,12 @@ def parse_arguments():
     )
 
     # Workload
-    # parser.add_argument(
-    #     "--jobs",
-    #     type=int,
-    #     metavar="<int>",
-    #     default=int(np.floor(cpu_count() / 2)),
-    #     help="Max number of concurrent jobs.",
-    # )
     parser.add_argument(
         "--cpu_budget",
         type=int,
         metavar="<int>",
         default=int(np.floor(cpu_count() / 2)),
-        help="Number of cpus to be used per hour.",
+        help="Number of cpus to be used.",
     )
     parser.add_argument(
         "--openroad_threads",

tools/AutoTuner/test/smoke_test_sweep.py

@@ -28,15 +28,17 @@ def setUp(self):
 
         # limit jobs because ray.get() does not terminate if jobs > number of samples
         core = os.cpu_count()
-        self.jobs = 4 if core >= 4 else core
+        self.resources_per_trial = 1
+        self.cpu_budget = 4 if core >= 4 else core
         self.experiment = f"smoke-test-sweep-{self.platform}"
         self.command = (
             "python3 -m autotuner.distributed"
             f" --design {self.design}"
             f" --platform {self.platform}"
             f" --experiment {self.experiment}"
             f" --config {self.config}"
-            f" --jobs {self.jobs}"
+            f" --cpu_budget {self.cpu_budget}"
+            f" --resources_per_trial {self.resources_per_trial}"
             f" sweep"
         )
 

tools/AutoTuner/test/test_timeout.py …ols/AutoTuner/test/smoke_test_timeout.pytools/AutoTuner/test/test_timeout.py renamed to tools/AutoTuner/test/smoke_test_timeout.py tools/AutoTuner/test/test_timeout.py renamed to tools/AutoTuner/test/smoke_test_timeout.py

@@ -16,7 +16,7 @@ def setUp(self):
             cur_dir,
             f"../../../flow/designs/{self.platform}/{self.design}/autotuner.json",
         )
-        self.experiment = f"smoke-test-tune-{self.platform}"
+        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"]