feat!: improved GPU job support

docs/further.md

@@ -118,8 +118,9 @@ You can use the following specifications:
 | `--cpus-per-task`       | `cpus_per_task`      | number of cpus per task (in case of SMP, rather use `threads`)   |
 | `--nodes` | `nodes`    | number of nodes                       |
 | `--clusters` | `clusters` | comma separated string of clusters |
+| `--gres`  | `gres` | generic resource specification, e.g. `gres=gpu:1`, allows for arbitray strings after the `=` |
 
-Each of these can be part of a rule, e.g.:
+Each of these can be part of a rule, e.g.: 
 
 ``` python
 rule:

snakemake_executor_plugin_slurm/__init__.py

@@ -158,6 +158,29 @@ def run_job(self, job: JobExecutorInterface):
         if self.workflow.executor_settings.requeue:
             call += " --requeue"
 
+        if job.resources.get("gres"):
+            # Validate GRES format (e.g., "gpu:1", "gpu:tesla:2")
+            gres = job.resources.gres
+            if not re.match(r"^[a-zA-Z0-9]+:([a-zA-Z0-9]+:)?\d+$", gres):
+                raise WorkflowError(
+                    f"Invalid GRES format: {gres}. Expected format: "
+                    "'<name>:<number>' or '<name>:<type>:<number>'"
+                )
+            call += f" --gres={job.resources.gres}"
+        if job.resources.get("gpu") and job.resources.get("gpu_model"):
+            # ensure that gres is not set, if gpu and gpu_model are set
+            if job.resources.get("gres"):
+                raise WorkflowError(
+                    "GRES and GPU model are set. Please only set one of them."
+                )
+            # ensure that 'gpu' is an integer
+            if not isinstance(job.resources.gpu, int):
+                raise WorkflowError(
+                    "The 'gpu' resource must be an integer. "
+                    f"Got: {job.resources.gpu} ({type(job.resources.gpu)})."
+                )
+            call += f" --gres=gpu:{job.resources.gpu_model}:{job.resources.gpu}"
+
         if job.resources.get("clusters"):
             call += f" --clusters {job.resources.clusters}"