mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity

Only include the tests if BUILD_TESTS is turned on.

Browse Source
This commit is contained in:
Nick Whitlock
2021-06-28 13:55:31 -04:00
parent c306b36860
commit acc1f70967
1 changed files with 54 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files
CMakeLists.txt
+54 -46
View File
@@ -24,6 +24,8 @@ else()
add_definitions(-DTKDNN_PATH="${CMAKE_CURRENT_SOURCE_DIR}") add_definitions(-DTKDNN_PATH="${CMAKE_CURRENT_SOURCE_DIR}")
endif() endif()
option(BUILD_TESTS "Whether to build the tests or not." ON)
#------------------------------------------------------------------------------- #-------------------------------------------------------------------------------
# CUDA # CUDA
@@ -72,62 +74,64 @@ target_link_libraries(tkDNN ${tkdnn_LIBS})
#add_library(tkDNN_static STATIC ${tkdnn_SRC}) #add_library(tkDNN_static STATIC ${tkdnn_SRC})
#target_link_libraries(tkDNN_static ${tkdnn_LIBS}) #target_link_libraries(tkDNN_static ${tkdnn_LIBS})
# SMALL NETS if(BUILD_TESTS)
add_executable(test_simple tests/simple/test_simple.cpp) # SMALL NETS
target_link_libraries(test_simple tkDNN) add_executable(test_simple tests/simple/test_simple.cpp)
target_link_libraries(test_simple tkDNN)
add_executable(test_mnist tests/mnist/test_mnist.cpp) add_executable(test_mnist tests/mnist/test_mnist.cpp)
target_link_libraries(test_mnist tkDNN) target_link_libraries(test_mnist tkDNN)
add_executable(test_mnistRT tests/mnist/test_mnistRT.cpp) add_executable(test_mnistRT tests/mnist/test_mnistRT.cpp)
target_link_libraries(test_mnistRT tkDNN) target_link_libraries(test_mnistRT tkDNN)
add_executable(test_imuodom tests/imuodom/imuodom.cpp) add_executable(test_imuodom tests/imuodom/imuodom.cpp)
target_link_libraries(test_imuodom tkDNN) target_link_libraries(test_imuodom tkDNN)
# DARKNET # DARKNET
file(GLOB darknet_SRC "tests/darknet/*.cpp") file(GLOB darknet_SRC "tests/darknet/*.cpp")
foreach(test_SRC ${darknet_SRC}) foreach(test_SRC ${darknet_SRC})
get_filename_component(test_NAME "${test_SRC}" NAME_WE) get_filename_component(test_NAME "${test_SRC}" NAME_WE)
set(test_NAME test_${test_NAME}) set(test_NAME test_${test_NAME})
add_executable(${test_NAME} ${test_SRC}) add_executable(${test_NAME} ${test_SRC})
target_link_libraries(${test_NAME} tkDNN) target_link_libraries(${test_NAME} tkDNN)
install(TARGETS ${test_NAME} DESTINATION bin) install(TARGETS ${test_NAME} DESTINATION bin)
endforeach() endforeach()
# MOBILENET # MOBILENET
add_executable(test_mobilenetv2ssd tests/mobilenet/mobilenetv2ssd/mobilenetv2ssd.cpp) add_executable(test_mobilenetv2ssd tests/mobilenet/mobilenetv2ssd/mobilenetv2ssd.cpp)
target_link_libraries(test_mobilenetv2ssd tkDNN) target_link_libraries(test_mobilenetv2ssd tkDNN)
add_executable(test_bdd-mobilenetv2ssd tests/mobilenet/bdd-mobilenetv2ssd/bdd-mobilenetv2ssd.cpp) add_executable(test_bdd-mobilenetv2ssd tests/mobilenet/bdd-mobilenetv2ssd/bdd-mobilenetv2ssd.cpp)
target_link_libraries(test_bdd-mobilenetv2ssd tkDNN) target_link_libraries(test_bdd-mobilenetv2ssd tkDNN)
add_executable(test_mobilenetv2ssd512 tests/mobilenet/mobilenetv2ssd512/mobilenetv2ssd512.cpp) add_executable(test_mobilenetv2ssd512 tests/mobilenet/mobilenetv2ssd512/mobilenetv2ssd512.cpp)
target_link_libraries(test_mobilenetv2ssd512 tkDNN) target_link_libraries(test_mobilenetv2ssd512 tkDNN)
# BACKBONES # BACKBONES
add_executable(test_resnet101 tests/backbones/resnet101/resnet101.cpp) add_executable(test_resnet101 tests/backbones/resnet101/resnet101.cpp)
target_link_libraries(test_resnet101 tkDNN) target_link_libraries(test_resnet101 tkDNN)
add_executable(test_dla34 tests/backbones/dla34/dla34.cpp) add_executable(test_dla34 tests/backbones/dla34/dla34.cpp)
target_link_libraries(test_dla34 tkDNN) target_link_libraries(test_dla34 tkDNN)
# CENTERNET # CENTERNET
add_executable(test_resnet101_cnet tests/centernet/resnet101_cnet/resnet101_cnet.cpp) add_executable(test_resnet101_cnet tests/centernet/resnet101_cnet/resnet101_cnet.cpp)
target_link_libraries(test_resnet101_cnet tkDNN) target_link_libraries(test_resnet101_cnet tkDNN)
add_executable(test_dla34_cnet tests/centernet/dla34_cnet/dla34_cnet.cpp) add_executable(test_dla34_cnet tests/centernet/dla34_cnet/dla34_cnet.cpp)
target_link_libraries(test_dla34_cnet tkDNN) target_link_libraries(test_dla34_cnet tkDNN)
# DEMOS # DEMOS
add_executable(test_rtinference tests/test_rtinference/rtinference.cpp) add_executable(test_rtinference tests/test_rtinference/rtinference.cpp)
target_link_libraries(test_rtinference tkDNN) target_link_libraries(test_rtinference tkDNN)
add_executable(map_demo demo/demo/map.cpp) add_executable(map_demo demo/demo/map.cpp)
target_link_libraries(map_demo tkDNN) target_link_libraries(map_demo tkDNN)
add_executable(demo demo/demo/demo.cpp) add_executable(demo demo/demo/demo.cpp)
target_link_libraries(demo tkDNN) target_link_libraries(demo tkDNN)
endif()
#------------------------------------------------------------------------------- #-------------------------------------------------------------------------------
# Install # Install
@@ -139,10 +143,14 @@ target_link_libraries(demo tkDNN)
message("install dir:" ${CMAKE_INSTALL_PREFIX}) message("install dir:" ${CMAKE_INSTALL_PREFIX})
install(DIRECTORY include/ DESTINATION include/) install(DIRECTORY include/ DESTINATION include/)
install(TARGETS tkDNN kernels DESTINATION lib) install(TARGETS tkDNN kernels DESTINATION lib)
install(TARGETS test_simple test_mnist test_mnistRT test_rtinference demo map_demo DESTINATION bin) if(BUILD_TESTS)
install(TARGETS test_simple test_mnist test_mnistRT test_rtinference demo map_demo DESTINATION bin)
endif()
install(DIRECTORY "${CMAKE_CURRENT_SOURCE_DIR}/cmake/" # source directory install(DIRECTORY "${CMAKE_CURRENT_SOURCE_DIR}/cmake/" # source directory
DESTINATION "share/tkDNN/cmake/" # target directory DESTINATION "share/tkDNN/cmake/" # target directory
) )
install(DIRECTORY "${CMAKE_CURRENT_SOURCE_DIR}/tests/" # source directory if(BUILD_TESTS)
DESTINATION "share/tkDNN/tests" # target directory install(DIRECTORY "${CMAKE_CURRENT_SOURCE_DIR}/tests/" # source directory
) DESTINATION "share/tkDNN/tests" # target directory
)
endif()