Skip to content

Commit e090656

Browse files
author
Thomas
committed
Changed configs
1 parent ae47cf9 commit e090656

File tree

1 file changed

+12
-3
lines changed

1 file changed

+12
-3
lines changed

pySDC/projects/GPU/analysis_scripts/parallel_scaling.py

Lines changed: 12 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
from pySDC.projects.GPU.configs.base_config import get_config
66
from pySDC.projects.GPU.etc.generate_jobscript import write_jobscript, PROJECT_PATH
77
from pySDC.helpers.plot_helper import setup_mpl, figsize_by_journal
8+
89
setup_mpl()
910

1011

@@ -32,14 +33,22 @@ def get_resolution_and_tasks(self, strong, i):
3233
if strong:
3334
return self.base_resolution, [1, self._tasks_time, 2**i]
3435
else:
35-
return self.base_resolution_weak * int(self._tasks_time**(1./self.ndim)) * (2**i), [1, self._tasks_time, (2 * self.ndim) ** i]
36+
return self.base_resolution_weak * int(self._tasks_time ** (1.0 / self.ndim)) * (2**i), [
37+
1,
38+
self._tasks_time,
39+
(2 * self.ndim) ** i,
40+
]
3641

3742
def run_scaling_test(self, strong=True):
3843
max_steps = self.max_steps_space if strong else self.max_steps_space_weak
3944
for i in range(max_steps):
4045
res, procs = self.get_resolution_and_tasks(strong, i)
4146

42-
sbatch_options = [f'-n {np.prod(procs)}', f'-p {self.partition}', f'--tasks-per-node={self.tasks_per_node}'] + self.sbatch_options
47+
sbatch_options = [
48+
f'-n {np.prod(procs)}',
49+
f'-p {self.partition}',
50+
f'--tasks-per-node={self.tasks_per_node}',
51+
] + self.sbatch_options
4352
srun_options = [f'--tasks-per-node={self.tasks_per_node}']
4453
if self.useGPU:
4554
srun_options += ['--cpus-per-task=4', '--gpus-per-task=1']
@@ -116,7 +125,7 @@ class GrayScottSpaceScalingGPU(GPUConfig, ScalingConfig):
116125
base_resolution = 8192
117126
config = 'GS_scaling'
118127
max_steps_space = 7
119-
max_steps_space_weak = 4
128+
max_steps_space_weak = 5
120129
tasks_time = 4
121130

122131

0 commit comments

Comments
 (0)