SDC22

Author: Thomas Baumann

pySDC/projects/GPU/etc/venv_jusuf/modules.sh

@@ -1,5 +1,5 @@
 module --force purge
-module load Stages/2024
+module load Stages/2025
 module load GCC
 module load ParaStationMPI
 # module load NCCL
@@ -9,6 +9,6 @@ module load Python
 # module load CuPy
 module load FFTW
 module load mpi4py
-module load FFmpeg/.6.0
+# module load FFmpeg/.6.0
 # module load SciPy-Stack
 module load texlive

pySDC/projects/GPU/etc/venv_jusuf/requirements.txt

@@ -3,3 +3,4 @@ numpy
 matplotlib>=3.0
 dill>=0.2.6
 scipy>=1.14
+tqdm

pySDC/projects/GPU/etc/venv_jusuf/setup.sh

@@ -12,7 +12,8 @@ python3 -m venv --prompt "$ENV_NAME" --system-site-packages "${ENV_DIR}"
 
 source "${ABSOLUTE_PATH}"/activate.sh
 
-FFTW_LIBRARY_DIR="/p/software/jusuf/stages/2024/software/FFTW/3.3.10-GCC-12.3.0/lib64" python3 -m pip install -e /p/project/ccstma/baumann7/mpi4py-fft
+
+FFTW_LIBRARY_DIR="/p/software/jusuf/stages/2025/software/FFTW/3.3.10-GCC-13.3.0/lib64/" python3 -m pip install -e /p/project1/ccstma/baumann7/mpi4py-fft
 python3 -m pip install -e /p/project1/ccstma/baumann7/qmat
 python3 -m pip install -r "${ABSOLUTE_PATH}"/requirements.txt
 python3 -m pip install -e /p/project1/ccstma/baumann7/pySDC/

pySDC/projects/RayleighBenard/RBC3D_configs.py

@@ -212,6 +212,15 @@ def get_description(self, *args, **kwargs):
         return desc
 
 
+class RBC3DM2K2(RBC3Dverification):
+
+    def get_description(self, *args, **kwargs):
+        desc = super().get_description(*args, **kwargs)
+        desc['level_params']['nsweeps'] = 2
+        desc['sweeper_params']['num_nodes'] = 2
+        return desc
+
+
 class RBC3DM3K4(RBC3Dverification):
 
     def get_description(self, *args, **kwargs):
@@ -256,9 +265,9 @@ def get_description(self, *args, res=-1, dt=-1, **kwargs):
         return desc
 
 
-class RBC3DG4R4Ra1e5(RBC3Dverification):
+class RBC3DG4R4SDC22Ra1e5(RBC3DM2K2):
     Tend = 200
-    dt = 6e-2
+    dt = 4e-2
     res = 32
     converged = 50
 

pySDC/projects/RayleighBenard/analysis_scripts/RBC3D_order.py

@@ -10,6 +10,7 @@
 from pySDC.projects.RayleighBenard.analysis_scripts.plotting_utils import get_plotting_style, savefig
 
 step_sizes = {
+    'RBC3DG4R4SDC22Ra1e5': [5e-3 * 2**i for i in range(8)],
     'RBC3DG4R4SDC23Ra1e5': [5e-3 * 2**i for i in range(8)],
     'RBC3DG4R4SDC34Ra1e5': [5e-3 * 2**i for i in range(8)],
     'RBC3DG4R4SDC44Ra1e5': [5e-3 * 2**i for i in range(8)],
@@ -115,7 +116,7 @@ def run(args, dt, Tend):
     prob = desc['problem_class'](**desc['problem_params'])
 
     ic_config_name = type(config).__name__
-    for name in ['RK', 'Euler', 'O3', 'O4', 'SDC23', 'SDC34', 'SDC44']:
+    for name in ['RK', 'Euler', 'O3', 'O4', 'SDC23', 'SDC34', 'SDC44', 'SDC22']:
         ic_config_name = ic_config_name.replace(name, 'SDC34')
 
     ic_config = get_config({**args, 'config': ic_config_name})

pySDC/projects/RayleighBenard/analysis_scripts/process_RBC3D_data.py

@@ -13,6 +13,7 @@ def process_RBC3D_data(base_path='./data/processed', plot=True, args=None, confi
     # prepare problem instance
     args = args if args else parse_args()
     comm = MPI.COMM_WORLD
+    args['procs'] = [1, 1, comm.size]
     config = config if config else get_config(args)
     desc = config.get_description(**args)
     P = desc['problem_class'](