controller: Add flag to en-/disable ram mgmt

Fraunhofer-AISEC · Mar 5, 2024 · d7e7b8e · d7e7b8e
1 parent a7615da
commit d7e7b8e
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/controller.py b/controller.py
@@ -729,8 +729,11 @@ def controller(
             break
 
         if (
-            mem_limit_calc(mem_max, len(p_list), queue_output.qsize(), time_max)
-            < max_ram
+            (
+                not args.enable_ram_mgmt
+                or mem_limit_calc(mem_max, len(p_list), queue_output.qsize(), time_max)
+                < max_ram
+            )
             and len(p_list) < num_workers
             and itter < len(faultlist)
             and queue_output.qsize() < queuedepth
@@ -921,6 +924,12 @@ def get_argument_parser():
         action="store_true",
         required=False,
     )
+    parser.add_argument(
+        "--enable-ram-mgmt",
+        help="Use with caution, may lead to drastic performance decrease",
+        action="store_true",
+        required=False,
+    )
     return parser