diff --git a/apex/contrib/csrc/group_norm_v2/gn_cuda_kernel.cuh b/apex/contrib/csrc/group_norm_v2/gn_cuda_kernel.cuh index 05f2c9854..366bfbb08 100644 --- a/apex/contrib/csrc/group_norm_v2/gn_cuda_kernel.cuh +++ b/apex/contrib/csrc/group_norm_v2/gn_cuda_kernel.cuh @@ -2,6 +2,8 @@ #include +#include + #include "gn_utils.hpp" namespace group_norm_v2 {