Skip to content

GitLab

"...git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "5321f3e2035675ee3f749fae2298a2bc6a6f012a"
Unverified Commit 183d29de authored by Jinjing Zhou's avatar Jinjing Zhou Committed by GitHub
Browse files

fix cub problem (#3121)

parent 2f41fcd9
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/array/cuda/rowwise_sampling.cu
View file @ 183d29de
...@@ -7,9 +7,9 @@ ...@@ -7,9 +7,9 @@
#include <dgl/random.h> #include <dgl/random.h>
#include <dgl/runtime/device_api.h> #include <dgl/runtime/device_api.h>
#include <curand_kernel.h> #include <curand_kernel.h>
#include <cub/cub.cuh>
#include <numeric> #include <numeric>
#include "./dgl_cub.cuh"
#include "../../kernel/cuda/atomic.cuh" #include "../../kernel/cuda/atomic.cuh"
#include "../../runtime/cuda/cuda_common.h" #include "../../runtime/cuda/cuda_common.h"
......
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