example: disable test if too few CPU

scivision · Jul 24, 2023 · c180ae7 · c180ae7
1 parent 16c140a
commit c180ae7
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 0 deletions.
diff --git a/example/CMakeLists.txt b/example/CMakeLists.txt
@@ -36,6 +36,19 @@ end program"
 scalapack_s
 )
 
+cmake_host_system_information(RESULT Ncpu QUERY NUMBER_OF_PHYSICAL_CORES)
+if(Ncpu LESS 6)
+  set(less6 true)
+endif()
+if(Ncpu LESS 4)
+  set(less4 true)
+endif()
+if(Ncpu LESS 2)
+  set(less2 true)
+endif()
+
+message(STATUS "Number of physical cores: ${Ncpu}")
+
 
 if(scalapack_d)
   add_executable(test_pdgesv test_pdgesv.f)
@@ -44,13 +57,23 @@ if(scalapack_d)
   COMMAND ${MPIEXEC_EXECUTABLE} ${MPIEXEC_NUMPROC_FLAG} 6 $<TARGET_FILE:test_pdgesv>
   )
   set_property(TEST PDGESV PROPERTY PROCESSORS 6)
+  set_property(TEST PDGESV PROPERTY DISABLED $<BOOL:${less6}>)
 
   add_executable(test_pdsyev test_pdsyev.f)
   target_link_libraries(test_pdsyev PRIVATE SCALAPACK::SCALAPACK LAPACK::LAPACK MPI::MPI_Fortran)
   add_test(NAME PDSYEV
   COMMAND ${MPIEXEC_EXECUTABLE} ${MPIEXEC_NUMPROC_FLAG} 4 $<TARGET_FILE:test_pdsyev>
   )
   set_property(TEST PDSYEV PROPERTY PROCESSORS 4)
+  set_property(TEST PDSYEV PROPERTY DISABLED $<BOOL:${less4}>)
+
+  add_executable(test_scalapack_d ${PROJECT_SOURCE_DIR}/../test/test_scalapack_d.f90)
+  target_link_libraries(test_scalapack_d PRIVATE SCALAPACK::SCALAPACK)
+
+  add_test(NAME ScalapackReal64
+  COMMAND ${MPIEXEC_EXECUTABLE} ${MPIEXEC_NUMPROC_FLAG} 2 $<TARGET_FILE:test_scalapack_d>
+  )
+  set_property(TEST ScalapackReal64 PROPERTY DISABLED $<BOOL:${less2}>)
 endif()
 
 if(scalapack_s)
@@ -60,6 +83,7 @@ if(scalapack_s)
   COMMAND ${MPIEXEC_EXECUTABLE} ${MPIEXEC_NUMPROC_FLAG} 4 $<TARGET_FILE:test_pssyev>
   )
   set_property(TEST PSSYEV PROPERTY PROCESSORS 4)
+  set_property(TEST PSSYEV PROPERTY DISABLED $<BOOL:${less4}>)
 endif()
 
 get_property(test_names DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR} PROPERTY TESTS)

diff --git a/test/CMakeLists.txt b/test/CMakeLists.txt
@@ -16,7 +16,15 @@ if(BUILD_DOUBLE)
   )
 endif()
 
+cmake_host_system_information(RESULT Ncpu QUERY NUMBER_OF_PHYSICAL_CORES)
+if(Ncpu LESS 2)
+  set(less2 true)
+endif()
+message(STATUS "Number of physical cores: ${Ncpu}")
+
+
 get_property(test_names DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR} PROPERTY TESTS)
 
 set_property(TEST ${test_names} PROPERTY TIMEOUT 15)
 set_property(TEST ${test_names} PROPERTY PROCESSORS 2)
+set_property(TEST ${test_names} PROPERTY DISABLED $<BOOL:${less2}>)