Merge pull request #83 from zenotech/ProblemNomenclatureChange

gres support
zenotech · Jul 14, 2023 · 55574b8 · 55574b8
2 parents 42f5413 + de33f60
commit 55574b8
Show file tree

Hide file tree

Showing 6 changed files with 10 additions and 0 deletions.
diff --git a/mycluster/schedulers/base.py b/mycluster/schedulers/base.py
@@ -104,6 +104,7 @@ def create_submit(
         qos=None,
         exclusive=True,
         output_name=None,
+        gres=None,
     ):
         """
         Write a new job file

diff --git a/mycluster/schedulers/lsf.py b/mycluster/schedulers/lsf.py
@@ -133,6 +133,7 @@ def create_submit(
         qos=None,
         exclusive=True,
         output_name=None,
+        gres=None
     ):
         if tasks_per_node is None:
             tasks_per_node = self.tasks_per_node(queue_id)

diff --git a/mycluster/schedulers/pbs.py b/mycluster/schedulers/pbs.py
@@ -154,6 +154,7 @@ def create_submit(
         qos=None,
         exclusive=True,
         output_name=None,
+        gres=None
     ):
         queue_name = queue_id
         if tasks_per_node is None:

diff --git a/mycluster/schedulers/sge.py b/mycluster/schedulers/sge.py
@@ -255,6 +255,7 @@ def create_submit(
         qos=None,
         exclusive=True,
         output_name=None,
+        gres=None
     ):
         parallel_env = queue_id.split(":")[0]
         queue_name = queue_id.split(":")[1]

diff --git a/mycluster/schedulers/slurm.py b/mycluster/schedulers/slurm.py
@@ -124,6 +124,7 @@ def create_submit(
         qos=None,
         exclusive=True,
         output_name=None,
+        gres=None
     ):
         queue_name = queue_id
         if tasks_per_node is None:
@@ -158,6 +159,7 @@ def create_submit(
             openmpi_args=openmpi_args,
             qos=qos,
             exclusive=exclusive,
+            gres=gres
         )
         return script_str
 

diff --git a/mycluster/schedulers/templates/slurm.jinja b/mycluster/schedulers/templates/slurm.jinja
@@ -31,6 +31,10 @@
 {% if qos %}
 #SBATCH --qos {{qos}}
 {% endif %}
+{% if gres %}
+#SBATCH --gres={{gres}}
+{% endif %}
+
 
 export MYCLUSTER_QUEUE={{queue_name}}
 export MYCLUSTER_JOB_NAME={{my_name}}