Add option to change number of ranks for tests

erf-model · Jan 4, 2024 · e842f59 · e842f59
1 parent cf993eb
commit e842f59
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/Tests/CMakeLists.txt b/Tests/CMakeLists.txt
@@ -1,3 +1,4 @@
 
 set(FCOMPARE_EXE ${CMAKE_BINARY_DIR}/Submodules/AMReX/Tools/Plotfile/amrex_fcompare CACHE INTERNAL "Path to fcompare executable for regression tests")
+set(ERF_TEST_NRANKS 4 CACHE STRING  "Number of MPI ranks to use for each test")
 include(${CMAKE_CURRENT_SOURCE_DIR}/CTestList.cmake)
diff --git a/Tests/CTestList.cmake b/Tests/CTestList.cmake
@@ -18,7 +18,7 @@ macro(setup_test)
     file(COPY ${TEST_FILES} DESTINATION "${CURRENT_TEST_BINARY_DIR}/")
 
     if(ERF_ENABLE_MPI)
-        set(NP 4)
+        set(NP ${ERF_TEST_NRANKS})
         set(MPI_COMMANDS "${MPIEXEC_EXECUTABLE} ${MPIEXEC_NUMPROC_FLAG} ${NP} ${MPIEXEC_PREFLAGS}")
         set(MPI_FCOMP_COMMANDS "${MPIEXEC_EXECUTABLE} ${MPIEXEC_NUMPROC_FLAG} 1 ${MPIEXEC_PREFLAGS}")
     else()