Implement max pooling test

2025-12-22 22:34:22 +00:00 · 2024-03-20 21:44:04 +01:00
parent c062e89972
commit dfff0360d9
9 changed files with 134 additions and 32 deletions
--- a/test/CMakeLists.txt
+++ b/test/CMakeLists.txt
@@ -1,15 +1,15 @@
 find_package(GTest REQUIRED)
 include_directories(${GTEST_INCLUDE_DIRS})

+file(GLOB_RECURSE TEST_SOURCES
+    *.cu
+    kernels/*.cu
+    layers/*.cu
+)
+
 add_executable(test_main 
    EXCLUDE_FROM_ALL
-    layers/test_activation.cu
-    layers/test_concat.cu
-    layers/test_conv2d.cu
-    layers/test_dense.cu
-    layers/test_input.cu
-    kernels/test_activation_functions.cu
-    kernels/test_matmul.cu
+    ${TEST_SOURCES}
 )

 target_link_libraries(test_main ${GTEST_BOTH_LIBRARIES} CUDANet)