lala

Author: Thomas Baumann

pySDC/projects/GPU/analysis_scripts/parallel_scaling.py

@@ -297,6 +297,11 @@ class RayleighBenardDedalusComparison(CPUConfig, RBCBaseConfig):
     config = 'RBC_Tibo'
     max_steps_space = 6
     tasks_time = 4
+    experiments = [
+        Experiment(res=256, PinT=False, start=1, stop=64, marker='.'),
+        Experiment(res=256, PinT=True, start=4, stop=256, marker='.'),
+    ]
+
 
 
 class RayleighBenardDedalusComparisonGPU(GPUConfig, ScalingConfig):
@@ -336,6 +341,8 @@ def plot_scalings(problem, **kwargs):  # pragma: no cover
     ideal_lines = {
         ('GS3D', 'throughput'): {'x': [0.25, 400], 'y': [5e6, 8e9]},
         ('GS3D', 'time'): {'x': [0.25, 400], 'y': [80, 5e-2]},
+        ('RBC', 'throughput'): {'x': [1/10, 64], 'y': [2e4, 2e4*640]},
+        ('RBC', 'time'): {'x': [1/10, 64], 'y': [60, 60/640]},
     }
 
     fig, ax = plt.subplots(figsize=figsize_by_journal('TUHH_thesis', 1, 0.6))
@@ -383,7 +390,7 @@ def plot_scalings(problem, **kwargs):  # pragma: no cover
             config_classes += [GrayScottSpaceScalingGPU3D]
     elif args.problem == 'RBC':
         if args.XPU == 'CPU':
-            config_classes += [RayleighBenardSpaceScalingCPU, RayleighBenardSpaceScalingCPULarge]
+            config_classes += [RayleighBenardSpaceScalingCPU]
         else:
             config_classes += [RayleighBenardSpaceScalingGPU]
     elif args.problem == 'RBC_dedalus':

pySDC/projects/GPU/analysis_scripts/plot_large_simulations.py

@@ -54,7 +54,7 @@ def get_fig(self, *args, **kwargs):
 
     def save_fig(self, fig, name):
         path = f'plots/{self.name}_{name}.pdf'
-        fig.savefig(path, bbox_inches='tight')
+        fig.savefig(path, bbox_inches='tight', dpi=300)
         print(f'Saved {path!r}', flush=True)
 
     @property
@@ -281,9 +281,11 @@ def plot_series(self, test=False):
         if test:
             indices = [0, 1, 2, 3, 4, 5]
             process = 0
+            layer = 0
         else:
-            indices = [0, 10, 20, 30, 40, 91]
-            process = 96
+            indices = [91] # [0, 10, 20, 30, 40, 91]
+            process = 120# 141#20#96  # 11
+            layer = 6
 
         from mpl_toolkits.axes_grid1 import make_axes_locatable
 
@@ -300,14 +302,19 @@ def plot_series(self, test=False):
         for frame in frame_range:
 
             plt.rcParams['figure.constrained_layout.use'] = True
-            fig, ax = plt.subplots(1, 1, figsize=figsize_by_journal('TUHH_thesis', 0.4, 1.1))
+            fig, ax = plt.subplots(1, 1, figsize=figsize_by_journal('TUHH_thesis', 0.7, 1.1))
             divider = make_axes_locatable(ax)
             cax = divider.append_axes('right', size='3%', pad=0.03)
 
             path = self.get_path(idx=frame, n_space=process)
             with open(path, 'rb') as file:
                 data = pickle.load(file)
-            im = ax.pcolormesh(X[1][0], X[2][0], data['u'][0], cmap='binary', rasterized=True)
+
+            # im = ax.pcolormesh(X[1][0], X[2][0], np.max(data['v'], axis=0), cmap='binary', rasterized=True, vmin=0, vmax=0.5)
+            # im = ax.pcolormesh(X[1][layer], X[2][layer], data['v'][layer], cmap='binary', rasterized=True, vmin=0, vmax=0.5)
+            im = ax.pcolormesh(X[1][layer], X[2][layer], data['v'][layer], cmap='binary', rasterized=True, vmin=0, vmax=0.5)
+            ax.set_xlim((9, 14))
+            ax.set_ylim((-4, 1))
 
             fig.colorbar(im, cax)  # , label=f'$T(t={{{t:.1f}}})$')