mirror of
https://github.com/pjreddie/darknet.git
synced 2023-08-10 21:13:14 +03:00
41 lines
974 B
CMake
41 lines
974 B
CMake
# Config file for the Darknet package
|
|
|
|
get_filename_component(Darknet_CMAKE_DIR "${CMAKE_CURRENT_LIST_FILE}" PATH)
|
|
list(APPEND CMAKE_MODULE_PATH "${Darknet_CMAKE_DIR}")
|
|
|
|
include(CMakeFindDependencyMacro)
|
|
|
|
if(@ENABLE_OPENCV@)
|
|
find_dependency(OpenCV)
|
|
add_definitions(-DOPENCV)
|
|
endif()
|
|
|
|
find_dependency(Stb)
|
|
|
|
if(@ENABLE_CUDA@)
|
|
add_definitions(-DGPU)
|
|
if(@ENABLE_CUDNN@)
|
|
find_dependency(CUDNN)
|
|
add_definitions(-DCUDNN)
|
|
if(@CUDA_COMPUTE_MODEL@ VERSION_GREATER_EQUAL @CUDA_COMPUTE_MODEL_70@)
|
|
add_definitions(-DCUDNN_HALF)
|
|
message(STATUS "Enabled CUDNN_HALF")
|
|
endif()
|
|
endif()
|
|
endif()
|
|
|
|
set(CMAKE_THREAD_PREFER_PTHREAD ON)
|
|
find_dependency(Threads)
|
|
|
|
if(MSVC)
|
|
find_dependency(PThreads_windows)
|
|
endif()
|
|
|
|
if(@ENABLE_OPENMP@)
|
|
find_dependency(OpenMP)
|
|
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${OpenMP_CXX_FLAGS}")
|
|
endif()
|
|
|
|
# Our library dependencies (contains definitions for IMPORTED targets)
|
|
include("${Darknet_CMAKE_DIR}/DarknetTargets.cmake")
|