From bfc109368b0a8c5d644efc5b1c4ca6bdb2f2b06a Mon Sep 17 00:00:00 2001 From: Lin Guo Date: Wed, 8 Jan 2025 23:47:23 -0800 Subject: [PATCH] Add var def for `mpi_command`, serving as a default --- .../workflow_managers/slurm/workflow_manager.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/var/ramble/repos/builtin/workflow_managers/slurm/workflow_manager.py b/var/ramble/repos/builtin/workflow_managers/slurm/workflow_manager.py index 82513b983..d5cdd5af5 100644 --- a/var/ramble/repos/builtin/workflow_managers/slurm/workflow_manager.py +++ b/var/ramble/repos/builtin/workflow_managers/slurm/workflow_manager.py @@ -58,6 +58,18 @@ def __init__(self, file_path): description="hostlist variable used by various modifiers", ) + workflow_manager_variable( + name="srun_args", + default="-n {n_ranks}", + description="Arguments supplied to srun", + ) + + workflow_manager_variable( + name="mpi_command", + default="srun {srun_args}", + description="mpirun prefix, mostly served as an overridable default", + ) + register_template( name="batch_submit", src_name="batch_submit.tpl",