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"