Use CUDA 11 cub when available (#49)

mphowardlab · Apr 12, 2021 · c9d75cd · c9d75cd
1 parent d54fe88
commit c9d75cd
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 0 deletions.
diff --git a/azplugins/MPCDReversePerturbationFlowGPU.cu b/azplugins/MPCDReversePerturbationFlowGPU.cu
@@ -13,7 +13,12 @@
 #include <thrust/sort.h>
 #include <thrust/device_vector.h>
 #include <thrust/functional.h>
+
+#if __CUDACC_VER_MAJOR__ >= 11
+#include <cub/cub.cuh>
+#else
 #include "hoomd/extern/cub/cub/cub.cuh"
+#endif
 
 namespace azplugins
 {

diff --git a/azplugins/ParticleEvaporatorGPU.cu b/azplugins/ParticleEvaporatorGPU.cu
@@ -9,7 +9,12 @@
  */
 
 #include "ParticleEvaporatorGPU.cuh"
+
+#if __CUDACC_VER_MAJOR__ >= 11
+#include <cub/cub.cuh>
+#else
 #include "hoomd/extern/cub/cub/cub.cuh"
+#endif
 
 namespace azplugins
 {

diff --git a/azplugins/ReversePerturbationFlowGPU.cu b/azplugins/ReversePerturbationFlowGPU.cu
@@ -13,7 +13,12 @@
 #include <thrust/sort.h>
 #include <thrust/device_vector.h>
 #include <thrust/functional.h>
+
+#if __CUDACC_VER_MAJOR__ >= 11
+#include <cub/cub.cuh>
+#else
 #include "hoomd/extern/cub/cub/cub.cuh"
+#endif
 
 namespace azplugins
 {