Blame view

3rdparty/opencv-4.5.4/modules/dnn/src/cuda4dnn/csl/nvcc_defs.hpp 632 Bytes
f4334277   Hu Chunming   提交3rdparty
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
  // 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.
  
  #ifndef OPENCV_DNN_SRC_CUDA4DNN_CSL_NVCC_DEFS_HPP
  #define OPENCV_DNN_SRC_CUDA4DNN_CSL_NVCC_DEFS_HPP
  
  #include <cuda_runtime_api.h>
  
  #ifdef __CUDACC__
  #   define CUDA4DNN_HOST __host__
  #   define CUDA4DNN_DEVICE __device__
  #   define CUDA4DNN_HOST_DEVICE CUDA4DNN_HOST CUDA4DNN_DEVICE
  #else
  #   define CUDA4DNN_HOST
  #   define CUDA4DNN_DEVICE
  #   define CUDA4DNN_HOST_DEVICE
  #endif
  
  #endif /* OPENCV_DNN_SRC_CUDA4DNN_CSL_NVCC_DEFS_HPP */