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