diff --git a/CMakeLists.txt b/CMakeLists.txt index 8c8619d..06bcb2d 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -113,6 +113,14 @@ target_link_libraries(map_demo tkDNN) add_executable(demo demo/demo/demo.cpp) target_link_libraries(demo tkDNN) +option(BUILD_SHARED_LIBS "Specifies the type of libraries (SHARED or STATIC) to build" ON) +if (BUILD_SHARED_LIBS) + add_library(darknetTR SHARED demo/demo/darknetTR.cpp) + target_compile_definitions(darknetTR PRIVATE LIB_EXPORTS=1) + target_compile_definitions(darknetTR PRIVATE -DDEMO_EXPORTS) + target_link_libraries(darknetTR tkDNN) +endif() + #------------------------------------------------------------------------------- # Install #-------------------------------------------------------------------------------