first draft packing nodes

lsst · Dec 5, 2023 · 6336508 · 6336508
1 parent e3d61e8
commit 6336508
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/python/lsst/ctrl/execute/allocatorParser.py b/python/lsst/ctrl/execute/allocatorParser.py
@@ -148,6 +148,13 @@ def parseArgs(self, basename):
             default=None,
             help="glide-in inactivity shutdown time in seconds",
         )
+        parser.add_argument(
+            "-p",
+            "--pack",
+            action="store_true",
+            dest="packnodes",
+            help="encouraging nodes to pack jobs rather than spread",
+        )
         parser.add_argument(
             "-v", "--verbose", action="store_true", dest="verbose", help="verbose"
         )

diff --git a/python/lsst/ctrl/execute/slurmPlugin.py b/python/lsst/ctrl/execute/slurmPlugin.py
@@ -137,6 +137,11 @@ def loadSlurm(self, name, platformPkgDir):
             self.allocationFileName
         )
 
+        if self.opts.packnodes is None:
+            self.defaults["PACK_BLOCK"] = "#"
+        else:
+            self.defaults["PACK_BLOCK"] = "Rank = TotalCpus - Cpus"          
+
         # handle dynamic slot block template:
         # 1) if it isn't specified, just put a comment in it's place
         # 2) if it's specified, but without a filename, use the default