fix: wfbench.py gets mem in megabytes

wfcommons · Nov 20, 2023 · 43ca19d · 43ca19d
1 parent c0ede14
commit 43ca19d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/wfcommons/wfbench/bench.py b/wfcommons/wfbench/bench.py
@@ -162,7 +162,7 @@ def create_benchmark_from_synthetic_workflow(
             )
             task.cores = task_cores + 1
             if task_memory:
-                task.memory = task_memory
+                task.memory = task_memory * 1024 * 1024  # megabytes to bytes
 
         # create data footprint
         for task in self.workflow.tasks.values():

diff --git a/wfcommons/wfbench/wfbench.py b/wfcommons/wfbench/wfbench.py
@@ -195,8 +195,10 @@ def main():
 
     print(f"[WfBench] Starting {args.name} Benchmark\n")
 
+    mem_bytes = args.mem * 1024 * 1024 if args.mem else None
+
     if args.out:
-        io_read_benchmark_user_input_data_size(other, memory_limit=args.mem)
+        io_read_benchmark_user_input_data_size(other, memory_limit=mem_bytes)
 
     if args.gpu_work:
         print("[WfBench] Starting GPU Benchmark...")
@@ -234,7 +236,7 @@ def main():
 
     if args.out:
         outputs = json.loads(args.out.replace("'", '"'))
-        io_write_benchmark_user_input_data_size(outputs, memory_limit=args.mem)
+        io_write_benchmark_user_input_data_size(outputs, memory_limit=mem_bytes)
 
     if core:
         unlock_core(path_locked, path_cores, core)