diff --git a/core/modes/slurm/shared.py b/core/modes/slurm/shared.py
index 50166d338225d7978bd61bb2e0fb978f8103209a..7c4e9f6ca9ecc4da297a66f9daa58706dc664b06 100644
--- a/core/modes/slurm/shared.py
+++ b/core/modes/slurm/shared.py
@@ -69,6 +69,8 @@ class SingleSlurmJobAcrossStages(Slurm, ABC):
             additional_env.append(f"{k}={v}")
         if len(additional_env) != 0:
             parameters += f' --export=' + ",".join(additional_env)
+        if self.slurm_job_id:
+            parameters += f' --jobid={self.slurm_job_id}'
         return parameters
 
     def get_simple_run_wrapper(self):
diff --git a/utility/.gitlab/.template.yml b/utility/.gitlab/.template.yml
index 78a6160e2456bf7ba7b342847dcfefcaa431d299..97e339a3ed543b90c39f0ef62261a2036e990630 100644
--- a/utility/.gitlab/.template.yml
+++ b/utility/.gitlab/.template.yml
@@ -47,6 +47,7 @@ variables:
   variables:
     SLURM_PARAM_CPUS: "-c 2"
     CI_MODE: "SingleSlurmJobAcrossStages"
+    SLURM_PARAM_GPUS: "--gres=gpu:volta:2"
   parallel:
     matrix:
       - PARVAR_X: ['a', 'b']
@@ -64,7 +65,7 @@ variables:
 
 .partition-check:
   variables:
-    _PARTITION: "c18m"
+    _PARTITION: "c18g"
     SLURM_PARAM_PARTITION: "-p ${_PARTITION}"
   script:
     - echo "SLURM_JOB_PARTITION ${SLURM_JOB_PARTITION}"
@@ -137,6 +138,8 @@ single-build-job:
   extends: 
     - .sequential-job
     - .partition-check
+  variables:
+    _PARTITION: "c18m"
   script:
     - echo "JOBID ${SLURM_JOB_ID}"
     - echo "Building on $(hostname) into $TMP"