From 37f25c15858808f9cb8697f8d6462edecadb2886 Mon Sep 17 00:00:00 2001
From: Quentin-Anthony <qganthony@yahoo.com>
Date: Tue, 28 Mar 2023 20:52:04 +0000
Subject: [PATCH] add slurm label

---
 deepspeed/launcher/multinode_runner.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/deepspeed/launcher/multinode_runner.py b/deepspeed/launcher/multinode_runner.py
index 959f76a4fc58..f7196f5bd8ba 100644
--- 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', ''):