mirror of https://github.com/opencv/opencv.git
Merge pull request #14660 from YashasSamaga:dnn-cuda-build
add cuDNN dependency and setup build for cuda4dnn (#14660) * update cmake for cuda4dnn - Adds FindCUDNN - Adds new options: * WITH_CUDA * OPENCV_DNN_CUDA - Adds CUDA4DNN preprocessor symbol for the DNN module * FIX: append EXCLUDE_CUDA instead of overwrite * remove cuDNN dependency for user apps * fix unused variable warningpull/14696/head
Yashas Samaga B L
6 years ago
committed by
Alexander Alekhin
parent
3ccb1d67c2
commit
ae279966c2
13 changed files with 228 additions and 11 deletions
@ -0,0 +1,105 @@ |
||||
# template taken from https://cmake.org/cmake/help/v3.14/manual/cmake-developer.7.html |
||||
|
||||
# Distributed under the OSI-approved BSD 3-Clause License. See accompanying |
||||
# file Copyright.txt or https://cmake.org/licensing for details. |
||||
|
||||
#[=======================================================================[.rst: |
||||
FindCUDNN |
||||
--------- |
||||
|
||||
Finds the cuDNN library. |
||||
|
||||
Requires: |
||||
^^^^^^^^^ |
||||
|
||||
find_cuda_helper_libs from FindCUDA.cmake |
||||
i.e. CUDA module should be found using FindCUDA.cmake before attempting to find cuDNN |
||||
|
||||
Result Variables |
||||
^^^^^^^^^^^^^^^^ |
||||
|
||||
This will define the following variables: |
||||
|
||||
``CUDNN_FOUND`` |
||||
``CUDNN_INCLUDE_DIRS`` location of cudnn.h |
||||
``CUDNN_LIBRARIES`` location of cudnn library |
||||
|
||||
Cache Variables |
||||
^^^^^^^^^^^^^^^ |
||||
|
||||
The following cache variables will be set if cuDNN was found. They may also be set on failure. |
||||
|
||||
``CUDNN_LIBRARY`` |
||||
``CUDNN_INCLUDE_DIR`` |
||||
``CUDNN_VERSION`` |
||||
|
||||
``CUDNN_VERSION_MAJOR`` INTERNAL |
||||
``CUDNN_VERSION_MINOR`` INTERNAL |
||||
``CUDNN_VERSION_PATCH`` INTERNAL |
||||
|
||||
#]=======================================================================] |
||||
|
||||
# find the library |
||||
if(CUDA_FOUND) |
||||
find_cuda_helper_libs(cudnn) |
||||
set(CUDNN_LIBRARY ${CUDA_cudnn_LIBRARY} CACHE FILEPATH "location of the cuDNN library") |
||||
unset(CUDA_cudnn_LIBRARY CACHE) |
||||
endif() |
||||
|
||||
# find the include |
||||
if(CUDNN_LIBRARY) |
||||
find_path(CUDNN_INCLUDE_DIR |
||||
cudnn.h |
||||
PATHS ${CUDA_TOOLKIT_INCLUDE} |
||||
DOC "location of cudnn.h" |
||||
NO_DEFAULT_PATH |
||||
) |
||||
|
||||
if(NOT CUDNN_INCLUDE_DIR) |
||||
find_path(CUDNN_INCLUDE_DIR |
||||
cudnn.h |
||||
DOC "location of cudnn.h" |
||||
) |
||||
endif() |
||||
endif() |
||||
|
||||
# extract version from the include |
||||
if(CUDNN_INCLUDE_DIR) |
||||
file(READ "${CUDNN_INCLUDE_DIR}/cudnn.h" CUDNN_H_CONTENTS) |
||||
|
||||
string(REGEX MATCH "define CUDNN_MAJOR ([0-9]+)" _ "${CUDNN_H_CONTENTS}") |
||||
set(CUDNN_MAJOR_VERSION ${CMAKE_MATCH_1} CACHE INTERNAL "") |
||||
string(REGEX MATCH "define CUDNN_MINOR ([0-9]+)" _ "${CUDNN_H_CONTENTS}") |
||||
set(CUDNN_MINOR_VERSION ${CMAKE_MATCH_1} CACHE INTERNAL "") |
||||
string(REGEX MATCH "define CUDNN_PATCHLEVEL ([0-9]+)" _ "${CUDNN_H_CONTENTS}") |
||||
set(CUDNN_PATCH_VERSION ${CMAKE_MATCH_1} CACHE INTERNAL "") |
||||
|
||||
set(CUDNN_VERSION |
||||
"${CUDNN_MAJOR_VERSION}.${CUDNN_MINOR_VERSION}.${CUDNN_PATCH_VERSION}" |
||||
CACHE |
||||
STRING |
||||
"cuDNN version" |
||||
) |
||||
|
||||
unset(CUDNN_H_CONTENTS) |
||||
endif() |
||||
|
||||
include(FindPackageHandleStandardArgs) |
||||
find_package_handle_standard_args(CUDNN |
||||
FOUND_VAR CUDNN_FOUND |
||||
REQUIRED_VARS |
||||
CUDNN_LIBRARY |
||||
CUDNN_INCLUDE_DIR |
||||
VERSION_VAR CUDNN_VERSION |
||||
) |
||||
|
||||
if(CUDNN_FOUND) |
||||
set(CUDNN_LIBRARIES ${CUDNN_LIBRARY}) |
||||
set(CUDNN_INCLUDE_DIRS ${CUDNN_INCLUDE_DIR}) |
||||
endif() |
||||
|
||||
mark_as_advanced( |
||||
CUDNN_LIBRARY |
||||
CUDNN_INCLUDE_DIR |
||||
CUDNN_VERSION |
||||
) |
||||
@ -1,3 +1,11 @@ |
||||
message(STATUS "opencv_dnn: filter out ocl4dnn source code") |
||||
ocv_list_filterout(OPENCV_MODULE_${the_module}_SOURCES "/ocl4dnn/") |
||||
ocv_list_filterout(OPENCV_MODULE_${the_module}_HEADERS "/ocl4dnn/") |
||||
if(NOT (OPENCV_DNN_OPENCL AND HAVE_OPENCL)) |
||||
message(STATUS "opencv_dnn: filter out ocl4dnn source code") |
||||
ocv_list_filterout(OPENCV_MODULE_${the_module}_SOURCES "/ocl4dnn/") |
||||
ocv_list_filterout(OPENCV_MODULE_${the_module}_HEADERS "/ocl4dnn/") |
||||
endif() |
||||
|
||||
if(NOT (OPENCV_DNN_CUDA AND HAVE_CUDA AND HAVE_CUBLAS AND HAVE_CUDNN)) |
||||
message(STATUS "opencv_dnn: filter out cuda4dnn source code") |
||||
ocv_list_filterout(OPENCV_MODULE_${the_module}_SOURCES "/cuda4dnn/") |
||||
ocv_list_filterout(OPENCV_MODULE_${the_module}_HEADERS "/cuda4dnn/") |
||||
endif() |
||||
|
||||
@ -0,0 +1,18 @@ |
||||
// This file is part of OpenCV project. |
||||
// It is subject to the license terms in the LICENSE file found in the top-level directory |
||||
// of this distribution and at http://opencv.org/license.html. |
||||
|
||||
// this file is a stub and will be removed once actual code is added |
||||
|
||||
#include "../precomp.hpp" |
||||
|
||||
#include <cuda_runtime.h> |
||||
|
||||
#ifndef HAVE_CUDA |
||||
# error "CUDA files should not be compiled if CUDA was not enabled" |
||||
#endif |
||||
|
||||
__global__ void cuda4dnn_build_test_kernel(float* addr) { |
||||
int idx = threadIdx.x; |
||||
addr[idx] = 0.0; |
||||
} |
||||
@ -0,0 +1,18 @@ |
||||
// This file is part of OpenCV project.
|
||||
// It is subject to the license terms in the LICENSE file found in the top-level directory
|
||||
// of this distribution and at http://opencv.org/license.html.
|
||||
|
||||
// this file is a stub and will be removed once actual code is added
|
||||
|
||||
#include "../precomp.hpp" |
||||
|
||||
#ifndef HAVE_CUDA |
||||
# error "CUDA4DNN should be enabled iff CUDA and cuDNN were found" |
||||
#endif |
||||
|
||||
#include <cudnn.h> |
||||
|
||||
void cuda4dnn_build_test_func() { |
||||
auto ver = cudnnGetVersion(); |
||||
CV_UNUSED(ver); |
||||
} |
||||
Loading…
Reference in new issue