move resources_per_trial to sweep_parser

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

Author: luarss

tools/AutoTuner/src/autotuner/distributed.py

@@ -234,7 +234,7 @@ def parse_arguments():
         help="mode of execution", dest="mode", required=True
     )
     tune_parser = subparsers.add_parser("tune")
-    _ = subparsers.add_parser("sweep")
+    sweep_parser = subparsers.add_parser("sweep")
 
     # DUT
     parser.add_argument(
@@ -343,7 +343,7 @@ def parse_arguments():
         help="Additional arguments given to ./build_openroad.sh.",
     )
 
-    # ML
+    # Tune mode
     tune_parser.add_argument(
         "--algorithm",
         type=str,
@@ -401,6 +401,14 @@ def parse_arguments():
         help="Random seed. (0 means no seed.)",
     )
 
+    sweep_parser.add_argument(
+        "--resources_per_trial",
+        type=int,
+        metavar="<int>",
+        default=1,
+        help="Number of CPUs to request for each sweep job.",
+    )
+
     # Workload
     parser.add_argument(
         "--cpu_budget",

tools/AutoTuner/test/smoke_test_sweep.py

@@ -38,8 +38,7 @@ def setUp(self):
             f" --experiment {self.experiment}"
             f" --config {self.config}"
             f" --cpu_budget {self.cpu_budget}"
-            f" --resources_per_trial {self.resources_per_trial}"
-            f" sweep"
+            f" sweep --resources_per_trial {self.resources_per_trial}"
         )
 
     def test_sweep(self):