missing process_worker_pool --available-accelerators

Parsl · Mar 14, 2024 · 6584fe4 · 6584fe4
1 parent 53734b5
commit 6584fe4
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 4 deletions.
diff --git a/output.txt b/output.txt
@@ -0,0 +1 @@
+1
diff --git a/parsl/executors/high_throughput/process_worker_pool.py b/parsl/executors/high_throughput/process_worker_pool.py
@@ -892,9 +892,13 @@ def strategyorlist(s: str):
         logger.info("Heartbeat period: {}".format(args.hb_period))
         logger.info("Drain period: {}".format(args.drain_period))
         logger.info("CPU affinity: {}".format(args.cpu_affinity))
-        logger.info("Accelerators: {}".format(" ".join(args.available_accelerators)))
-        logger.info("enable_mpi_mode: {}".format(args.enable_mpi_mode))
-        logger.info("mpi_launcher: {}".format(args.mpi_launcher))
+        if isinstance(args.available_accelerators, list):
+            logger.info("Accelerators: {}".format(" ".join(args.available_accelerators)))
+        else:
+            logger.info("Accelerators: {}".format(args.available_accelerators))
+            # logger.info("Accelerators: {}".format(" ".join(args.available_accelerators)))
+            logger.info("enable_mpi_mode: {}".format(args.enable_mpi_mode))
+            logger.info("mpi_launcher: {}".format(args.mpi_launcher))
 
         manager = Manager(task_port=args.task_port,
                           result_port=args.result_port,
@@ -911,7 +915,8 @@ def strategyorlist(s: str):
                           prefetch_capacity=int(args.prefetch_capacity),
                           heartbeat_threshold=int(args.hb_threshold),
                           heartbeat_period=int(args.hb_period),
-                          drain_period=None if args.drain_period == "None" else int(args.drain_period),
+                          drain_period=None if args.drain_period is None or args.drain_period == "None" else int(args.drain_period),
+                          # drain_period=None if args.drain_period == "None" else int(args.drain_period),
                           poll_period=int(args.poll),
                           cpu_affinity=args.cpu_affinity,
                           enable_mpi_mode=args.enable_mpi_mode,

diff --git a/std.out b/std.out
@@ -0,0 +1 @@
+hello there