Mem per command was never used - fixed

Author: ddtm

scripts/smart-dispatch

@@ -131,7 +131,7 @@ def main():
 
     command_params = {'nb_cores_per_command': args.coresPerCommand,
                       'nb_gpus_per_command': args.gpusPerCommand,
-                      'mem_per_command': None  # args.memPerCommand
+                      'mem_per_command': args.memPerCommand
                       }
 
     job_generator = job_generator_factory(queue, commands, command_params, CLUSTER_NAME, path_job)

smartdispatch/job_generator.py

@@ -77,9 +77,6 @@ def _generate_base_pbs(self):
         # Limit number of running commands by the amount of available memory on the node.
         if self.mem_per_command is not None:
             nb_commands_per_node = min(nb_commands_per_node, self.queue.mem_per_node // self.mem_per_command)
-            mem_per_command = self.mem_per_command
-        else:
-            mem_per_command = self.queue.mem_per_node // nb_commands_per_node
 
         pbs_files = []
         # Distribute equally the jobs among the PBS files and generate those files
@@ -95,8 +92,9 @@ def _generate_base_pbs(self):
                 resource += ":gpus={gpus}".format(gpus=len(commands) * self.nb_gpus_per_command)
             pbs.add_resources(nodes=resource)
 
-            resource = "{mem}Gb".format(mem=len(commands) * mem_per_command)
-            pbs.add_resources(mem=resource)
+            if self.mem_per_command is not None:
+                resource = "{mem}Gb".format(mem=len(commands) * self.mem_per_command)
+                pbs.add_resources(mem=resource)
 
             pbs.add_modules_to_load(*self.queue.modules)
             pbs.add_commands(*commands)