From acc1f709670640db8e592310ddef14c5d62096d6 Mon Sep 17 00:00:00 2001
From: Nick Whitlock <nick.whitlock@techsource.com>
Date: Mon, 28 Jun 2021 13:55:31 -0400
Subject: [PATCH] Only include the tests if BUILD_TESTS is turned on.

---
 CMakeLists.txt | 100 ++++++++++++++++++++++++++-----------------------
 1 file changed, 54 insertions(+), 46 deletions(-)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index f03c98c3..862e3743 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -24,6 +24,8 @@ else()
     add_definitions(-DTKDNN_PATH="${CMAKE_CURRENT_SOURCE_DIR}")
 endif()
 
+option(BUILD_TESTS "Whether to build the tests or not." ON)
+
 
 #-------------------------------------------------------------------------------
 # CUDA
@@ -72,62 +74,64 @@ target_link_libraries(tkDNN ${tkdnn_LIBS})
 #add_library(tkDNN_static STATIC ${tkdnn_SRC})
 #target_link_libraries(tkDNN_static ${tkdnn_LIBS})
 
-# SMALL NETS
-add_executable(test_simple tests/simple/test_simple.cpp)
-target_link_libraries(test_simple tkDNN)
+if(BUILD_TESTS)
+	# SMALL NETS
+	add_executable(test_simple tests/simple/test_simple.cpp)
+	target_link_libraries(test_simple tkDNN)
 
-add_executable(test_mnist tests/mnist/test_mnist.cpp)
-target_link_libraries(test_mnist tkDNN)
+	add_executable(test_mnist tests/mnist/test_mnist.cpp)
+	target_link_libraries(test_mnist tkDNN)
 
-add_executable(test_mnistRT tests/mnist/test_mnistRT.cpp)
-target_link_libraries(test_mnistRT tkDNN)
+	add_executable(test_mnistRT tests/mnist/test_mnistRT.cpp)
+	target_link_libraries(test_mnistRT tkDNN)
 
-add_executable(test_imuodom tests/imuodom/imuodom.cpp)
-target_link_libraries(test_imuodom tkDNN)
+	add_executable(test_imuodom tests/imuodom/imuodom.cpp)
+	target_link_libraries(test_imuodom tkDNN)
 
-# DARKNET
-file(GLOB darknet_SRC "tests/darknet/*.cpp")
-foreach(test_SRC ${darknet_SRC})
-    get_filename_component(test_NAME "${test_SRC}" NAME_WE)
-    set(test_NAME test_${test_NAME})
-    add_executable(${test_NAME} ${test_SRC})
-    target_link_libraries(${test_NAME} tkDNN)
-    install(TARGETS ${test_NAME} DESTINATION bin)
-endforeach()
+	# DARKNET
+	file(GLOB darknet_SRC "tests/darknet/*.cpp")
+	foreach(test_SRC ${darknet_SRC})
+		get_filename_component(test_NAME "${test_SRC}" NAME_WE)
+		set(test_NAME test_${test_NAME})
+		add_executable(${test_NAME} ${test_SRC})
+		target_link_libraries(${test_NAME} tkDNN)
+		install(TARGETS ${test_NAME} DESTINATION bin)
+	endforeach()
 
-# MOBILENET
-add_executable(test_mobilenetv2ssd tests/mobilenet/mobilenetv2ssd/mobilenetv2ssd.cpp)
-target_link_libraries(test_mobilenetv2ssd tkDNN)
+	# MOBILENET
+	add_executable(test_mobilenetv2ssd tests/mobilenet/mobilenetv2ssd/mobilenetv2ssd.cpp)
+	target_link_libraries(test_mobilenetv2ssd tkDNN)
 
-add_executable(test_bdd-mobilenetv2ssd tests/mobilenet/bdd-mobilenetv2ssd/bdd-mobilenetv2ssd.cpp)
-target_link_libraries(test_bdd-mobilenetv2ssd tkDNN)
+	add_executable(test_bdd-mobilenetv2ssd tests/mobilenet/bdd-mobilenetv2ssd/bdd-mobilenetv2ssd.cpp)
+	target_link_libraries(test_bdd-mobilenetv2ssd tkDNN)
 
-add_executable(test_mobilenetv2ssd512 tests/mobilenet/mobilenetv2ssd512/mobilenetv2ssd512.cpp)
-target_link_libraries(test_mobilenetv2ssd512 tkDNN)
+	add_executable(test_mobilenetv2ssd512 tests/mobilenet/mobilenetv2ssd512/mobilenetv2ssd512.cpp)
+	target_link_libraries(test_mobilenetv2ssd512 tkDNN)
 
-# BACKBONES
-add_executable(test_resnet101 tests/backbones/resnet101/resnet101.cpp)
-target_link_libraries(test_resnet101 tkDNN)
+	# BACKBONES
+	add_executable(test_resnet101 tests/backbones/resnet101/resnet101.cpp)
+	target_link_libraries(test_resnet101 tkDNN)
 
-add_executable(test_dla34 tests/backbones/dla34/dla34.cpp)
-target_link_libraries(test_dla34 tkDNN)
+	add_executable(test_dla34 tests/backbones/dla34/dla34.cpp)
+	target_link_libraries(test_dla34 tkDNN)
 
-# CENTERNET
-add_executable(test_resnet101_cnet tests/centernet/resnet101_cnet/resnet101_cnet.cpp)
-target_link_libraries(test_resnet101_cnet tkDNN)
+	# CENTERNET
+	add_executable(test_resnet101_cnet tests/centernet/resnet101_cnet/resnet101_cnet.cpp)
+	target_link_libraries(test_resnet101_cnet tkDNN)
 
-add_executable(test_dla34_cnet tests/centernet/dla34_cnet/dla34_cnet.cpp)
-target_link_libraries(test_dla34_cnet tkDNN)
+	add_executable(test_dla34_cnet tests/centernet/dla34_cnet/dla34_cnet.cpp)
+	target_link_libraries(test_dla34_cnet tkDNN)
 
-# DEMOS
-add_executable(test_rtinference tests/test_rtinference/rtinference.cpp)
-target_link_libraries(test_rtinference tkDNN)
+	# DEMOS
+	add_executable(test_rtinference tests/test_rtinference/rtinference.cpp)
+	target_link_libraries(test_rtinference tkDNN)
 
-add_executable(map_demo demo/demo/map.cpp)
-target_link_libraries(map_demo tkDNN)
+	add_executable(map_demo demo/demo/map.cpp)
+	target_link_libraries(map_demo tkDNN)
 
-add_executable(demo demo/demo/demo.cpp)
-target_link_libraries(demo tkDNN)
+	add_executable(demo demo/demo/demo.cpp)
+	target_link_libraries(demo tkDNN)
+endif()
 
 #-------------------------------------------------------------------------------
 # Install
@@ -139,10 +143,14 @@ target_link_libraries(demo tkDNN)
 message("install dir:" ${CMAKE_INSTALL_PREFIX})
 install(DIRECTORY include/ DESTINATION include/)
 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
         DESTINATION "share/tkDNN/cmake/" # target directory
 )
-install(DIRECTORY "${CMAKE_CURRENT_SOURCE_DIR}/tests/" # source directory
-        DESTINATION "share/tkDNN/tests" # target directory
-)
+if(BUILD_TESTS)
+	install(DIRECTORY "${CMAKE_CURRENT_SOURCE_DIR}/tests/" # source directory
+    	    DESTINATION "share/tkDNN/tests" # target directory
+	)
+endif()