Update to fix the addition of -g erroneously (#361)

LLNL · May 26, 2021 · 788561d · 788561d
1 parent 2d9ad58
commit 788561d
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/maestrowf/interfaces/script/_flux/flux0_26_0.py b/maestrowf/interfaces/script/_flux/flux0_26_0.py
@@ -98,10 +98,11 @@ def parallelize(cls, procs, nodes=None, **kwargs):
             args.append("-c")
             args.append(str(kwargs["cores per task"]))
 
-        if "gpus" in kwargs:
-            ngpus = str(kwargs["gpus"])
+        ngpus = kwargs.get("gpus", 0)
+        if ngpus:
+            gpus = str(ngpus)
             args.append("-g")
-            args.append(ngpus)
+            args.append(gpus)
 
         # flux has additional arguments that can be passed via the '-o' flag.
         addtl = []