Fix jobid of shared slurm job mode

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):

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"