File tree Expand file tree Collapse file tree 1 file changed +8
-0
lines changed
pySDC/projects/GPU/analysis_scripts Expand file tree Collapse file tree 1 file changed +8
-0
lines changed Original file line number Diff line number Diff 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
106112class GPUConfig (ScalingConfig ):
107113 cluster = 'booster'
108114 partition = 'booster'
109115 tasks_per_node = 4
110116 useGPU = True
117+ max_nodes = 936
111118
112119
113120class 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
132140def plot_scalings (strong , problem , kwargs ): # pragma: no cover
You can’t perform that action at this time.
0 commit comments