Skip to content

Commit f61d328

Browse files
author
Thomas Baumann
committed
Implemented node limit in scaling
#!!!!!! WARNING: FLAKEHEAVEN FAILED !!!!!!: #:
1 parent 8aeefd5 commit f61d328

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

pySDC/projects/GPU/analysis_scripts/parallel_scaling.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@ class ScalingConfig(object):
2222
max_steps_space = None
2323
max_steps_space_weak = None
2424
sbatch_options = []
25+
max_nodes = 9999
2526

2627
def __init__(self, space_time_parallel):
2728
if space_time_parallel in ['False', False]:
@@ -44,6 +45,10 @@ def run_scaling_test(self, strong=True):
4445
for i in range(max_steps):
4546
res, procs = self.get_resolution_and_tasks(strong, i)
4647

48+
_nodes = np.prod(procs) // self.tasks_per_node
49+
if _nodes > self.max_nodes:
50+
break
51+
4752
sbatch_options = [
4853
f'-n {np.prod(procs)}',
4954
f'-p {self.partition}',
@@ -101,13 +106,15 @@ class CPUConfig(ScalingConfig):
101106
cluster = 'jusuf'
102107
partition = 'batch'
103108
tasks_per_node = 16
109+
max_nodes = 144
104110

105111

106112
class GPUConfig(ScalingConfig):
107113
cluster = 'booster'
108114
partition = 'booster'
109115
tasks_per_node = 4
110116
useGPU = True
117+
max_nodes = 936
111118

112119

113120
class GrayScottSpaceScalingCPU(CPUConfig, ScalingConfig):
@@ -127,6 +134,7 @@ class GrayScottSpaceScalingGPU(GPUConfig, ScalingConfig):
127134
max_steps_space = 7
128135
max_steps_space_weak = 5
129136
tasks_time = 4
137+
max_nodes = 64
130138

131139

132140
def plot_scalings(strong, problem, kwargs): # pragma: no cover

0 commit comments

Comments
 (0)