From 1d17e60374c3ae395a43ac3c855660e35cb143ee Mon Sep 17 00:00:00 2001
From: Gagik Hakobyan <hakgagik@yahoo.com>
Date: Fri, 28 Dec 2018 23:54:48 +0400
Subject: [PATCH] libdarknet.so shared object generation for python

---
 CMakeLists.txt | 20 +++++++++++++++-----
 1 file changed, 15 insertions(+), 5 deletions(-)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index 02e2de50..8b8ebbe0 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -3,7 +3,7 @@ cmake_minimum_required(VERSION 3.8)
 set(GPU ON)
 set(CUDNN ON)
 set(OPENCV ON)
-set(DEBUG ON)
+set(DEBUG OFF)
 
 #add_definitions(-DOPENMP)
 cmake_policy(SET CMP0054 NEW)
@@ -192,8 +192,7 @@ set_runtime(STATIC)
 add_library(darknet_cpp STATIC ${CPP_SOURCES})
 add_library(darknet_app STATIC ${APP_SOURCES})
 
-add_executable(${PROJECT_NAME} ${C_SOURCES} ${CMAKE_CURRENT_SOURCE_DIR}/examples/darknet.c)
-#CUDA_ADD_CUBLAS_TO_TARGET(${PROJECT_NAME})
+add_library(${PROJECT_NAME} SHARED ${C_SOURCES})
 
 target_link_libraries(${PROJECT_NAME}
     darknet_cpp
@@ -208,13 +207,14 @@ if (OPENCV)
 endif()
 
 if (GPU)
+    set(CUDA_NVCC_FLAGS ${CUDA_NVCC_FLAGS};-Xcompiler="-fPIC")
     cuda_add_library(darknet_cuda STATIC ${CUDA_SOURCES})
     target_link_libraries(${PROJECT_NAME}
 	darknet_cuda
-	cublas
-	curand
+	${CUDA_curand_LIBRARY}
 	${CUDA_LIBRARIES}
     )
+    CUDA_ADD_CUBLAS_TO_TARGET(${PROJECT_NAME})
 endif()
 
 if (CUDNN)
@@ -223,3 +223,13 @@ if (CUDNN)
     )
 endif()
 
+set(APP_NAME executable)
+add_executable(${APP_NAME} ${CMAKE_CURRENT_SOURCE_DIR}/examples/darknet.c)
+target_link_libraries(${APP_NAME} ${PROJECT_NAME})
+set_target_properties(${APP_NAME} PROPERTIES OUTPUT_NAME ${PROJECT_NAME})
+
+add_custom_target(link_target ALL
+    COMMAND ${CMAKE_COMMAND} -E create_symlink
+    ${PROJECT_SOURCE_DIR}/install/lib${PROJECT_NAME}.so
+    ${PROJECT_SOURCE_DIR}/lib${PROJECT_NAME}.so)
+