forked from opencv/opencv
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request opencv#14660 from YashasSamaga:dnn-cuda-build
add cuDNN dependency and setup build for cuda4dnn (opencv#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 warning
- Loading branch information
1 parent
3ccb1d6
commit ae27996
Showing
13 changed files
with
228 additions
and
11 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -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 | ||
| ) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
14 changes: 11 additions & 3 deletions
14
modules/dnn/cmake/hooks/INIT_MODULE_SOURCES_opencv_dnn.cmake
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -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() |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -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; | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -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); | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters