build each test file as a separate executable

Author: simonbyrne

test/CMakeLists.txt

@@ -47,7 +47,6 @@ set (test_sources
     00_sparse/Matmul.cu
     00_sparse/Matvec.cu
     00_sparse/Solve.cu
-    main.cu
 )
 
 # Some of <00_io> tests need csv files and binaries which all
@@ -85,30 +84,64 @@ endforeach()
 set(target_inc ${CMAKE_CURRENT_SOURCE_DIR}/include ${CMAKE_CURRENT_SOURCE_DIR}/../examples/ ${proprietary_inc_list})
 set(system_inc ${CUTLASS_INC} ${GTEST_INC_DIRS} ${pybind11_INCLUDE_DIR} ${PYTHON_INCLUDE_DIRS})
 
-set(all_test_srcs ${test_sources} ${proprietary_sources})
+# Function to create individual test executables
+function(create_test_executable test_file)
+    # Get the filename without path and extension for the target name
+    get_filename_component(test_name ${test_file} NAME_WE)
+    # Get the directory to make target names unique
+    get_filename_component(test_dir ${test_file} DIRECTORY)
+    string(REPLACE "/" "_" test_dir_clean ${test_dir})
+    set(target_name "test_${test_dir_clean}_${test_name}")
 
-add_executable(matx_test main.cu ${all_test_srcs})
+    # Create executable with main.cu and the specific test file
+    add_executable(${target_name} main.cu ${test_file})
 
-# Set all the flags/other properties
-set_property(TARGET matx_test PROPERTY ENABLE_EXPORTS 1)
+    # Set all the flags/other properties
+    set_property(TARGET ${target_name} PROPERTY ENABLE_EXPORTS 1)
 
-if (DEFINED cupy_PYTHON_PACKAGE)
-    target_compile_definitions(matx_test PRIVATE CUPY_INSTALLED)
-endif()
+    if (DEFINED cupy_PYTHON_PACKAGE)
+        target_compile_definitions(${target_name} PRIVATE CUPY_INSTALLED)
+    endif()
 
-if (MSVC)
-    target_compile_options(matx_test  PRIVATE /W4 /WX)
-else()
-    target_compile_options(matx_test PRIVATE ${WARN_FLAGS})
-    #target_compile_options(matx_test PRIVATE $<$<COMPILE_LANGUAGE:CUDA>:${WARN_FLAGS}>)
-    target_compile_options(matx_test PRIVATE $<$<COMPILE_LANGUAGE:CUDA>:${MATX_CUDA_FLAGS}>)
-endif()
+    if (MSVC)
+        target_compile_options(${target_name} PRIVATE /W4 /WX)
+    else()
+        target_compile_options(${target_name} PRIVATE ${WARN_FLAGS})
+        target_compile_options(${target_name} PRIVATE $<$<COMPILE_LANGUAGE:CUDA>:${MATX_CUDA_FLAGS}>)
+    endif()
 
-target_include_directories(matx_test PRIVATE "${target_inc}")
-target_include_directories(matx_test SYSTEM PRIVATE "${system_inc}")
-target_link_libraries(matx_test PRIVATE matx::matx) # Transitive properties
-target_link_libraries(matx_test PRIVATE ${NVSHMEM_LIBRARY} gtest)
+    target_include_directories(${target_name} PRIVATE "${target_inc}")
+    target_include_directories(${target_name} SYSTEM PRIVATE "${system_inc}")
+    target_link_libraries(${target_name} PRIVATE matx::matx) # Transitive properties
+    target_link_libraries(${target_name} PRIVATE ${NVSHMEM_LIBRARY} gtest)
 
+    # Register the test with CTest
+    add_test(NAME ${target_name} COMMAND ${target_name})
+    set_tests_properties(${target_name} PROPERTIES
+        WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR}
+    )
+endfunction()
+
+# Enable CTest
+enable_testing()
+
+# Create individual executables for each test file
+foreach(test_file ${test_sources})
+    create_test_executable(${test_file})
+endforeach()
+
+# Create individual executables for proprietary tests
+foreach(test_file ${proprietary_sources})
+    create_test_executable(${test_file})
+endforeach()
+
+# Number of test jobs to run in parallel
+set(CTEST_PARALLEL_JOBS 4)
+
+# Add a custom target to run CTest from the main build directory
 add_custom_target(test
-    DEPENDS matx_test
-    COMMAND ${CMAKE_CURRENT_BINARY_DIR}/matx_test)
+    COMMAND ${CMAKE_CTEST_COMMAND} -j${CTEST_PARALLEL_JOBS} --output-on-failure
+    WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR}
+    COMMENT "Running all MatX tests in parallel (${CTEST_PARALLEL_JOBS} cores)"
+    VERBATIM
+)