Split tests to cpu and cuda

2025-11-05 17:34:21 +00:00 · 2024-09-07 21:24:34 +02:00
parent f8220f0ec1
commit 591507ed21
16 changed files with 29 additions and 10 deletions
--- a/test/CMakeLists.txt
+++ b/test/CMakeLists.txt
@@ -1,17 +1,34 @@
 find_package(GTest REQUIRED)
 include_directories(${GTEST_INCLUDE_DIRS})

-file(GLOB_RECURSE TEST_SOURCES
-    *.cu
-    kernels/*.cu
-    layers/*.cu
+file(GLOB COMMON_TEST_SOURCES
+    common/*.hpp
    model/*.cpp
 )

-add_executable(test_main 
-    EXCLUDE_FROM_ALL
-    ${TEST_SOURCES}
-)
+if(USE_CUDA)
+    file(GLOB TEST_SOURCES
+        cuda/*.cu
+        cuda/kernels/*.cu
+        cuda/layers/*.cu
+        cuda/utils/*.cu
+    )
+    add_executable(test_main 
+        EXCLUDE_FROM_ALL
+        ${COMMON_TEST_SOURCES}
+        ${TEST_SOURCES}
+    )
+    set_target_properties(test_main PROPERTIES CUDA_SEPARABLE_COMPILATION ON)
+else()
+    file(GLOB TEST_SOURCES
+        cpu/*.cpp
+    )
+    add_executable(test_main 
+        EXCLUDE_FROM_ALL
+        ${COMMON_TEST_SOURCES}
+        ${TEST_SOURCES}
+    )
+endif()

 target_link_libraries(test_main ${GTEST_BOTH_LIBRARIES} CUDANet)

--- a/test/cuda/kernels/test_activation_functions.cu
+++ b/test/cuda/kernels/test_activation_functions.cu
--- a/test/cuda/kernels/test_matmul.cu
+++ b/test/cuda/kernels/test_matmul.cu
--- a/test/cuda/layers/test_activation.cu
+++ b/test/cuda/layers/test_activation.cu
--- a/test/cuda/layers/test_avg_pooling.cu
+++ b/test/cuda/layers/test_avg_pooling.cu
--- a/test/cuda/layers/test_batch_norm.cu
+++ b/test/cuda/layers/test_batch_norm.cu
--- a/test/cuda/layers/test_concat.cu
+++ b/test/cuda/layers/test_concat.cu
--- a/test/cuda/layers/test_conv2d.cu
+++ b/test/cuda/layers/test_conv2d.cu
--- a/test/cuda/layers/test_dense.cu
+++ b/test/cuda/layers/test_dense.cu
--- a/test/cuda/layers/test_input.cu
+++ b/test/cuda/layers/test_input.cu
--- a/test/cuda/layers/test_max_pooling.cu
+++ b/test/cuda/layers/test_max_pooling.cu
--- a/test/cuda/layers/test_output.cu
+++ b/test/cuda/layers/test_output.cu
--- a/test/cuda/utils/test_vector.cu
+++ b/test/cuda/utils/test_vector.cu
--- a/test/model/test_model.cpp
+++ b/test/model/test_model.cpp