fix cub problem (#3121)

183d29de · Jinjing Zhou · GitHub · 2f41fcd9 · 183d29de
Unverified Commit 183d29de authored Jul 08, 2021 by Jinjing Zhou Committed by GitHub Jul 08, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/array/cuda/rowwise_sampling.cu src/array/cuda/rowwise_sampling.cu +1 -1

No files found.
--- a/src/array/cuda/rowwise_sampling.cu
+++ b/src/array/cuda/rowwise_sampling.cu
@@ -7,9 +7,9 @@
 #include <dgl/random.h>
 #include <dgl/runtime/device_api.h>
 #include <curand_kernel.h>
-#include <cub/cub.cuh>
 #include <numeric>

+#include "./dgl_cub.cuh"
 #include "../../kernel/cuda/atomic.cuh"
 #include "../../runtime/cuda/cuda_common.h"