From 473566f37c18416773e2d0f57cbb1fff6cb955a8 Mon Sep 17 00:00:00 2001 From: Riccardo Balin Date: Wed, 18 Jan 2023 14:52:34 -0500 Subject: [PATCH] Set number of tasks for PALS mpiexec (#247) Correction to flag setting number of tasks for PalsMpiexecSettings [ committed by @rickybalin ] [ reviewed by @MattToast, @ashao ] --- smartsim/settings/mpirunSettings.py | 2 +- smartsim/settings/palsSettings.py | 8 ++++++++ 2 files changed, 9 insertions(+), 1 deletion(-) diff --git a/smartsim/settings/mpirunSettings.py b/smartsim/settings/mpirunSettings.py index 797da3534..9034e766a 100644 --- a/smartsim/settings/mpirunSettings.py +++ b/smartsim/settings/mpirunSettings.py @@ -35,4 +35,4 @@ "mpirunSettings will be deprecated; use mpiSettings instead.", DeprecationWarning, stacklevel=2 -) \ No newline at end of file +) diff --git a/smartsim/settings/palsSettings.py b/smartsim/settings/palsSettings.py index 0bb17e643..8282ad107 100644 --- a/smartsim/settings/palsSettings.py +++ b/smartsim/settings/palsSettings.py @@ -127,6 +127,14 @@ def set_cpu_binding_type(self, bind_type): """ self.run_args["cpu-bind"] = str(bind_type) + def set_tasks(self, tasks): + """Set the number of tasks + + :param tasks: number of total tasks to launch + :type tasks: int + """ + self.run_args["np"] = int(tasks) + def set_tasks_per_node(self, tasks_per_node): """Set the number of tasks per node