mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-08-06 14:36:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Addressing CUDA9 shfl deprecation

Browse Source
This commit is contained in:
Boris Fomitchev 2017-08-29 13:27:21 -07:00
parent 91ef0b9537
commit dde04d5d3e
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
modules/core/include/opencv2/core/cuda/warp_shuffle.hpp
View File

@ -51,6 +51,11 @@
namespace cv { namespace cuda { namespace device namespace cv { namespace cuda { namespace device
{ {
#if __CUDACC_VER_MAJOR__ >= 9
# define __shfl(x, y, z) __shfl_sync(0xFFFFFFFFU, x, y, z)
# define __shfl_up(x, y, z) __shfl_up_sync(0xFFFFFFFFU, x, y, z)
# define __shfl_down(x, y, z) __shfl_down_sync(0xFFFFFFFFU, x, y, z)
#endif
template <typename T> template <typename T>
__device__ __forceinline__ T shfl(T val, int srcLane, int width = warpSize) __device__ __forceinline__ T shfl(T val, int srcLane, int width = warpSize)
{ {
@ -148,6 +153,10 @@ namespace cv { namespace cuda { namespace device
} }
}}} }}}
# undef __shfl
# undef __shfl_up
# undef __shfl_down
//! @endcond //! @endcond
#endif // OPENCV_CUDA_WARP_SHUFFLE_HPP #endif // OPENCV_CUDA_WARP_SHUFFLE_HPP