From 70b8c369d46abf562b68e6b3ae03a0a554113a3b Mon Sep 17 00:00:00 2001 From: Michal Sojka Date: Tue, 6 Mar 2018 08:46:02 +0100 Subject: [PATCH] cmake: Find CUDA before OpenCV This way, CUDA is searched only once. OpenCV seems to not search it again, if it is already found. --- CMakeLists.txt | 23 +++++++++++------------ 1 file changed, 11 insertions(+), 12 deletions(-) diff --git a/CMakeLists.txt b/CMakeLists.txt index 1ab0f30..5606b64 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -43,6 +43,17 @@ IF(ASYNC AND NOT OPENMP) MESSAGE(STATUS "ASYNC") ENDIF() #ASYNC +IF(use_cuda) + find_package(CUDA REQUIRED) + + include_directories ( ${CUDA_INCLUDE_DIRS} ) + set(CUDA_SEPARABLE_COMPILATION ON) + set(CUDA_PROPAGATE_HOST_FLAGS OFF) + set(CUDA_HOST_COMPILER /usr/bin/g++) + list( APPEND CUDA_NVCC_FLAGS "-O3 --gpu-architecture compute_62 -std=c++11") + find_cuda_helper_libs(cufftw) +ENDIF() + FIND_PACKAGE( OpenCV REQUIRED ) link_directories ( ${OpenCV_LIB_DIR} ) MESSAGE(STATUS "OpenCV_LIB_DIR: ${OpenCV_LIB_DIR} ") @@ -64,18 +75,6 @@ include_directories(${CMAKE_SOURCE_DIR}/src/ ${CMAKE_SOURCE_DIR}/src/piotr_fhog include_directories(${CMAKE_SOURCE_DIR}/sse2neon) -IF(use_cuda) - find_package(CUDA REQUIRED) - - include_directories ( ${CUDA_INCLUDE_DIRS} ) - MESSAGE(STATUS "CUDA_INCLUDE_DIRS: ${CUDA_INCLUDE_DIRS}") - set(CUDA_SEPARABLE_COMPILATION ON) - set(CUDA_PROPAGATE_HOST_FLAGS OFF) - set(CUDA_HOST_COMPILER /usr/bin/g++) - list( APPEND CUDA_NVCC_FLAGS "-O3 --gpu-architecture compute_62 -std=c++11") - find_cuda_helper_libs(cufftw) -ENDIF() - add_subdirectory(src) IF(NOT use_cuda) -- 2.39.2