Adding CUB Version Check

Bug 2880957
NVIDIA · Mar 30, 2020 · 5911ab2 · 5911ab2
1 parent ebc90b0
commit 5911ab2
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/dependencies/cub b/dependencies/cub
diff --git a/thrust/system/cuda/config.h b/thrust/system/cuda/config.h
@@ -71,3 +71,10 @@
 #define THRUST_CUB_NS_PREFIX namespace thrust {   namespace cuda_cub {
 #define THRUST_CUB_NS_POSTFIX }  }
 
+#ifndef THRUST_IGNORE_CUB_VERSION_CHECK
+#include <thrust/version.h>
+#include <cub/version.cuh>
+#if THRUST_VERSION != CUB_VERSION
+#error Thrust version must match cub version exactly. Define THRUST_IGNORE_CUB_VERSION_CHECK to ignore.
+#endif
+#endif
+334 −0		CMakeLists.txt
+13 −0		cmake/AppendOptionIfAvailable.cmake
+1 −0		cmake/header_test.in
+20 −0		cmake/run_example.cmake
+8 −0		cmake/run_test.cmake
+3 −0		cub/grid/grid_even_share.cuh
+2 −0		cub/thread/thread_search.cuh
+11 −0		cub/util_namespace.cuh
+70 −0		cub/version.cuh