Skip to content

Commit

Permalink
Merge pull request #369 from argonne-lcf/theta
Browse files Browse the repository at this point in the history
updated theta launch params
  • Loading branch information
cms21 committed Jul 14, 2023
2 parents a37901d + ea919c6 commit 1d23048
Showing 1 changed file with 8 additions and 0 deletions.
8 changes: 8 additions & 0 deletions balsam/platform/app_run/theta.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,13 @@ def _pre_popen(self) -> None:
def _build_cmdline(self) -> str:
node_ids = [nid for nid in self._node_spec.node_ids]
nid_str = ",".join(map(str, node_ids))

launch_params = []
for k in self._launch_params.keys():
if k != "cpu_affinity":
launch_params.append(k)
launch_params.append(str(self._launch_params[k]))

cpu_affinity = self._launch_params.get("cpu_affinity", "none")
if cpu_affinity not in ["none", "depth"]:
cpu_affinity = "none"
Expand All @@ -31,6 +38,7 @@ def _build_cmdline(self) -> str:
self._threads_per_rank,
"-j",
self._threads_per_core,
*launch_params,
self._cmdline,
]
return " ".join(str(arg) for arg in args)

0 comments on commit 1d23048

Please sign in to comment.