Add CDP opt out mechanism

NVIDIA · Oct 27, 2022 · 5b06d18 · 5b06d18
1 parent 799cc47
commit 5b06d18
Showing 1 changed file with 9 additions and 2 deletions.
diff --git 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__