add slurm label

microsoft · Mar 28, 2023 · 37f25c1 · 37f25c1
1 parent a9d0263
commit 37f25c1
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/deepspeed/launcher/multinode_runner.py b/deepspeed/launcher/multinode_runner.py
@@ -253,6 +253,7 @@ def get_cmd(self, environment, active_resources):
  'srun',
  '-n',
  f'{total_process_count}',
+ '--label',
  ] + split(self.args.launcher_args)
 
  if getattr(self.args, 'slurm_comment', ''):