|
5 | 5 | from pySDC.projects.GPU.configs.base_config import get_config |
6 | 6 | from pySDC.projects.GPU.etc.generate_jobscript import write_jobscript, PROJECT_PATH |
7 | 7 | from pySDC.helpers.plot_helper import setup_mpl, figsize_by_journal |
| 8 | + |
8 | 9 | setup_mpl() |
9 | 10 |
|
10 | 11 |
|
@@ -32,14 +33,22 @@ def get_resolution_and_tasks(self, strong, i): |
32 | 33 | if strong: |
33 | 34 | return self.base_resolution, [1, self._tasks_time, 2**i] |
34 | 35 | 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 | + ] |
36 | 41 |
|
37 | 42 | def run_scaling_test(self, strong=True): |
38 | 43 | max_steps = self.max_steps_space if strong else self.max_steps_space_weak |
39 | 44 | for i in range(max_steps): |
40 | 45 | res, procs = self.get_resolution_and_tasks(strong, i) |
41 | 46 |
|
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 |
43 | 52 | srun_options = [f'--tasks-per-node={self.tasks_per_node}'] |
44 | 53 | if self.useGPU: |
45 | 54 | srun_options += ['--cpus-per-task=4', '--gpus-per-task=1'] |
@@ -116,7 +125,7 @@ class GrayScottSpaceScalingGPU(GPUConfig, ScalingConfig): |
116 | 125 | base_resolution = 8192 |
117 | 126 | config = 'GS_scaling' |
118 | 127 | max_steps_space = 7 |
119 | | - max_steps_space_weak = 4 |
| 128 | + max_steps_space_weak = 5 |
120 | 129 | tasks_time = 4 |
121 | 130 |
|
122 | 131 |
|
|
0 commit comments