support for account, qos in Slurm submission

python/lsst/ctrl/execute/allocator.py

@@ -96,6 +96,9 @@ def __init__(self, platform, opts, configuration, condorInfoFileName):
         self.commandLineDefaults["NODE_COUNT"] = self.opts.nodeCount
         self.commandLineDefaults["CPUS"] = self.opts.cpus
         self.commandLineDefaults["WALL_CLOCK"] = self.opts.maximumWallClock
+        self.commandLineDefaults["ACCOUNT"] = self.opts.account
+        self.commandLineDefaults["QOS"] = self.opts.qos
+        self.commandLineDefaults["MEMPERCORE"] = 4096
         self.commandLineDefaults["QUEUE"] = self.opts.queue
         self.load()
 
@@ -303,6 +306,12 @@ def getNodes(self):
         """
         return self.getParameter("NODE_COUNT")
 
+    def getMemoryPerCore(self):
+        """Accessor for MemoryPerCore
+        @return the value of MemoryPerCore
+        """
+        return self.getParameter("MEMPERCORE")
+
     def getCPUs(self):
         """Accessor for CPUS
         @return the value of CPUS

python/lsst/ctrl/execute/allocatorParser.py

@@ -81,6 +81,24 @@ def parseArgs(self, basename):
             type=int,
             required=True,
         )
+        parser.add_argument(
+            "-a",
+            "--account",
+            action="store",
+            default="rubin:developers",
+            dest="account",
+            help="Slurm account for glidein job",
+            type=str,
+        )
+        parser.add_argument(
+            "-s",
+            "--qos",
+            action="store",
+            default="normal",
+            dest="qos",
+            help="Slurm qos or glidein job",
+            type=str,
+        )
         parser.add_argument(
             "-m",
             "--maximum-wall-clock",

python/lsst/ctrl/execute/slurmPlugin.py

@@ -61,6 +61,9 @@ def submit(self, platform, platformPkgDir):
         self.createAllocationFile(allocationName)
 
         nodes = self.getNodes()
+        cpus = self.getCPUs()
+        memoryPerCore = self.getMemoryPerCore()
+        totalMemory = cpus * memoryPerCore
         print("Targeting %s glidein(s) for the computing pool/set." % nodes)
 
         # run the sbatch command
@@ -77,7 +80,7 @@ def submit(self, platform, platformPkgDir):
 
             print("The generated Slurm submit file is %s " % generatedSlurmFile)
 
-        cmd = "sbatch %s" % generatedSlurmFile
+        cmd = "sbatch --mem %s %s" % (totalMemory, generatedSlurmFile)
 
         auser = self.getUserName()
         jobname = "".join(["glide_", auser])