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

533d0c46 · Frank Lee · GitHub · c27ea0d9 · 533d0c46
Unverified Commit 533d0c46 authored May 18, 2022 by Frank Lee Committed by GitHub May 18, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 2 deletions

colossalai/kernel/cuda_native/csrc/kernels/dropout_kernels.cu ...ssalai/kernel/cuda_native/csrc/kernels/dropout_kernels.cu +3 -2

No files found.
--- a/colossalai/kernel/cuda_native/csrc/kernels/dropout_kernels.cu
+++ b/colossalai/kernel/cuda_native/csrc/kernels/dropout_kernels.cu
-#include <cooperative_groups.h>
-
 #include <chrono>
 #include <ctime>

 #include "kernels.h"

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

 curandStatePhilox4_32_10_t *curandstate;