at: add arg to set memory limit

For reference, nangate45/gcd uses about ~3GB

Signed-off-by: Vitor Bandeira <[email protected]>

Author: vvbandeira

tools/AutoTuner/src/autotuner/distributed.py

@@ -380,6 +380,13 @@ def parse_arguments():
         default=42,
         help="Random seed. (0 means no seed.)",
     )
+    tune_parser.add_argument(
+        "--memory_limit",
+        type=float,
+        metavar="<float>",
+        default=1,
+        help="Maximum memory in GB that each trial job can use, process will be killed and not retried if it exceeds.",
+    )
 
     # Workload
     parser.add_argument(

tools/AutoTuner/src/autotuner/utils.py

@@ -324,7 +324,7 @@ def openroad(
     export_command += " && "
 
     make_command = export_command
-    make_command += f"make -C {base_dir}/flow DESIGN_CONFIG=designs/"
+    make_command += f"ulimit -v {args.memory_limit * 1_000_000}; make -C {base_dir}/flow DESIGN_CONFIG=designs/"
     make_command += f"{args.platform}/{args.design}/config.mk"
     make_command += f" PLATFORM={args.platform}"
     make_command += f" FLOW_VARIANT={flow_variant} {parameters}"