Skip to content

Commit ba146fb

Browse files
committed
Set max to available nodes
1 parent 933bb2b commit ba146fb

File tree

2 files changed

+5
-4
lines changed

2 files changed

+5
-4
lines changed

lib/pavilion/schedulers/advanced.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -608,6 +608,10 @@ def _schedule_indi_flex(self, pav_cfg, tests: List[TestRun],
608608

609609
node_range = calc_node_range(sched_config, len(chunk))
610610

611+
if node_range[1] is not None:
612+
if len(chunk) < node_range[1]:
613+
node_range = (node_range[0], len(chunk))
614+
611615
job_name = 'pav_{}'.format(test.name)
612616
script = self._create_kickoff_script_stub(
613617
pav_cfg=pav_cfg,

lib/pavilion/schedulers/plugins/slurm.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -73,12 +73,9 @@ def _kickoff_lines(self) -> List[str]:
7373
# The job is defined by # of tasks.
7474
if self._node_min != 1:
7575
lines.append('#SBATCH --nodes {}'.format(self._node_min))
76-
elif self._node_min != self._node_max:
77-
# Specify a node range.
78-
lines.append('#SBATCH --nodes {}-{}'.format(self._node_min, self._node_max))
7976
else:
8077
# Specify the minimum number of nodes.
81-
lines.append('#SBATCH --nodes {}'.format(self._node_min))
78+
lines.append('#SBATCH --nodes {}'.format(self._node_max))
8279

8380
tasks = self._config['tasks']
8481
if tasks is not None:

0 commit comments

Comments
 (0)