[kernel] fixed the include bug in dropout kernel (#999)

2025-09-17 07:00:37 +00:00 · 2022-05-18 21:43:18 +08:00
parent c27ea0d980
commit 533d0c46d8
1 changed files with 3 additions and 2 deletions
--- 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 <cooperative_groups.h>
-
 #include <chrono>
 #include <ctime>

 #include "kernels.h"

+#include <cooperative_groups.h>
+
+
 namespace cg = cooperative_groups;

 curandStatePhilox4_32_10_t *curandstate;