Update cub for cuda 11.5 compatibility (dmlc#3468)

ammuaj · Nov 3, 2021 · f510214 · f510214
1 parent cac25f6
commit f510214
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/array/cuda/dgl_cub.cuh b/src/array/cuda/dgl_cub.cuh
@@ -10,7 +10,9 @@
 // include cub in a safe manner
 #define CUB_NS_PREFIX namespace dgl {
 #define CUB_NS_POSTFIX }
+#define CUB_NS_QUALIFIER ::dgl::cub
 #include "cub/cub.cuh"
+#undef CUB_NS_QUALIFIER
 #undef CUB_NS_POSTFIX
 #undef CUB_NS_PREFIX