diff --git a/src/madengine/deployment/slurm.py b/src/madengine/deployment/slurm.py index f5fb2495..a5fe7640 100644 --- a/src/madengine/deployment/slurm.py +++ b/src/madengine/deployment/slurm.py @@ -676,6 +676,7 @@ def debug(self, msg): "network_interface": self.slurm_config.get("network_interface"), "exclusive": self.slurm_config.get("exclusive", True), "exclude": self.slurm_config.get("exclude"), + "reservation": self.slurm_config.get("reservation"), "constraint": self.slurm_config.get("constraint"), "nodelist": self._normalize_nodelist(self.slurm_config.get("nodelist")), "qos": self.slurm_config.get("qos"), diff --git a/src/madengine/deployment/templates/slurm/job.sh.j2 b/src/madengine/deployment/templates/slurm/job.sh.j2 index 5f8e8266..3b236d9f 100644 --- a/src/madengine/deployment/templates/slurm/job.sh.j2 +++ b/src/madengine/deployment/templates/slurm/job.sh.j2 @@ -8,6 +8,9 @@ #SBATCH --ntasks-per-node=1 #SBATCH --gpus-per-node={{ gpus_per_node }} #SBATCH --time={{ time_limit }} +{% if reservation %} +#SBATCH --reservation={{ reservation }} +{% endif %} {% if nodelist %} #SBATCH --nodelist={{ nodelist }} {% endif %}