Skip to content

Commit

Permalink
Add var def for mpi_command, serving as a default
Browse files Browse the repository at this point in the history
  • Loading branch information
linsword13 committed Jan 9, 2025
1 parent 6258d50 commit bfc1093
Showing 1 changed file with 12 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand Down

0 comments on commit bfc1093

Please sign in to comment.