diff --git a/colossalai/kernel/cuda_native/csrc/kernels/dropout_kernels.cu b/colossalai/kernel/cuda_native/csrc/kernels/dropout_kernels.cu index 184106bd2..a39a6dae0 100644 --- a/colossalai/kernel/cuda_native/csrc/kernels/dropout_kernels.cu +++ b/colossalai/kernel/cuda_native/csrc/kernels/dropout_kernels.cu @@ -1,10 +1,11 @@ -#include - #include #include #include "kernels.h" +#include + + namespace cg = cooperative_groups; curandStatePhilox4_32_10_t *curandstate;