Skip to content

GitLab

Unverified Commit f5102145 authored by nv-dlasalle's avatar nv-dlasalle Committed by GitHub
Browse files

Update cub for cuda 11.5 compatibility (#3468)

parent cac25f63
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
src/array/cuda/dgl_cub.cuh
View file @ f5102145
...@@ -10,7 +10,9 @@ ...@@ -10,7 +10,9 @@
// include cub in a safe manner // include cub in a safe manner
#define CUB_NS_PREFIX namespace dgl { #define CUB_NS_PREFIX namespace dgl {
#define CUB_NS_POSTFIX } #define CUB_NS_POSTFIX }
#define CUB_NS_QUALIFIER ::dgl::cub
#include "cub/cub.cuh" #include "cub/cub.cuh"
#undef CUB_NS_QUALIFIER
#undef CUB_NS_POSTFIX #undef CUB_NS_POSTFIX
#undef CUB_NS_PREFIX #undef CUB_NS_PREFIX
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment