mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-11-06 01:34:22 +00:00
15 lines
372 B
CMake
15 lines
372 B
CMake
find_package(GTest REQUIRED)
|
|
include_directories(${GTEST_INCLUDE_DIRS})
|
|
|
|
add_executable(test_main
|
|
layers/test_dense.cu
|
|
layers/test_conv2d.cu
|
|
layers/test_input.cu
|
|
kernels/test_activations.cu
|
|
kernels/test_padding.cu
|
|
kernels/test_matmul.cu
|
|
)
|
|
|
|
target_link_libraries(test_main ${GTEST_BOTH_LIBRARIES} CUDANet)
|
|
|
|
add_test(NAME TestMain COMMAND test_main) |