diff --git a/apex/contrib/csrc/group_norm_v2/gn_cuda_host_template.cuh b/apex/contrib/csrc/group_norm_v2/gn_cuda_host_template.cuh index 873239a21..c306516df 100644 --- a/apex/contrib/csrc/group_norm_v2/gn_cuda_host_template.cuh +++ b/apex/contrib/csrc/group_norm_v2/gn_cuda_host_template.cuh @@ -4,8 +4,11 @@ #include #include +#include #include #include +#include +#include #include "gn_cuda_kernel.cuh" #include "gn_utils.hpp"