From c9d75cd1a07b9e0e11746d07f3c286e9ceb1d3ec Mon Sep 17 00:00:00 2001
From: Michael Howard <mphoward@utexas.edu>
Date: Mon, 12 Apr 2021 12:44:44 -0500
Subject: [PATCH] Use CUDA 11 cub when available (#49)

---
 azplugins/MPCDReversePerturbationFlowGPU.cu | 5 +++++
 azplugins/ParticleEvaporatorGPU.cu          | 5 +++++
 azplugins/ReversePerturbationFlowGPU.cu     | 5 +++++
 3 files changed, 15 insertions(+)

diff --git a/azplugins/MPCDReversePerturbationFlowGPU.cu b/azplugins/MPCDReversePerturbationFlowGPU.cu
index 80f1f16c..4b3de19b 100644
--- 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
index 936f1948..3be3c93a 100644
--- 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
index 9c2b4ef2..d5aebfea 100644
--- 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
 {