diff --git a/cub/detail/detect_cuda_runtime.cuh b/cub/detail/detect_cuda_runtime.cuh index 7ac947277d..2ca962eae0 100644 --- a/cub/detail/detect_cuda_runtime.cuh +++ b/cub/detail/detect_cuda_runtime.cuh @@ -43,10 +43,17 @@ namespace detail #ifdef DOXYGEN_SHOULD_SKIP_THIS // Only parse this during doxygen passes: +/** + * \def CUB_DISABLE_CDP + * + * If defined, support for device-side usage of CUB is disabled. + */ +#define CUB_DISABLE_CDP + /** * \def CUB_RDC_ENABLED * - * Defined if RDC is enabled. + * Defined if RDC is enabled and CUB_DISABLE_CDP is not defined. */ #define CUB_RDC_ENABLED @@ -76,7 +83,7 @@ namespace detail #ifndef CUB_RUNTIME_FUNCTION -#if defined(__CUDACC_RDC__) +#if defined(__CUDACC_RDC__) && !defined(CUB_DISABLE_CDP) #define CUB_RDC_ENABLED #define CUB_RUNTIME_FUNCTION __host__ __device__