Bash_alias does not work. Testing replacement in file

Parallel-in-Time · Apr 24, 2024 · d8416c9 · d8416c9
1 parent f7ddca4
commit d8416c9
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pySDC/tests/test_sweepers/test_MPI_sweeper.py b/pySDC/tests/test_sweepers/test_MPI_sweeper.py
@@ -93,7 +93,10 @@ def individual_test(num_nodes, quad_type, residual_type, imex, initGuess, useNCC
         my_env['PYTHONPATH'] = '../../..:.'
         my_env['COVERAGE_PROCESS_START'] = 'pyproject.toml'
 
-        cmd = f"mpirun -np {num_nodes} python {__file__} --test_sweeper {num_nodes} {quad_type} {residual_type} {imex} {initGuess} {useNCCL}".split()
+        if os.environ.get('SYSTEMNAME') == "juwels":
+            cmd = f"srun -n {num_nodes} python {__file__} --test_sweeper {num_nodes} {quad_type} {residual_type} {imex} {initGuess} {useNCCL}".split()
+        else:
+            cmd = f"mpirun -np {num_nodes} python {__file__} --test_sweeper {num_nodes} {quad_type} {residual_type} {imex} {initGuess} {useNCCL}".split()
 
         p = subprocess.Popen(cmd, env=my_env, cwd=".")