diff --git a/projects/hip/include/hip/hip_cooperative_groups.h b/projects/hip/include/hip/hip_cooperative_groups.h index 5f6e18ea56..41f36378bb 100644 --- a/projects/hip/include/hip/hip_cooperative_groups.h +++ b/projects/hip/include/hip/hip_cooperative_groups.h @@ -30,12 +30,15 @@ THE SOFTWARE. #ifndef HIP_INCLUDE_HIP_HIP_COOPERATIVE_GROUP_H #define HIP_INCLUDE_HIP_HIP_COOPERATIVE_GROUP_H +#include +#include + #if defined(__HIP_PLATFORM_HCC__) && !defined(__HIP_PLATFORM_NVCC__) #if __cplusplus && defined(__clang__) && defined(__HIP__) #include #endif #elif defined(__HIP_PLATFORM_NVCC__) && !defined(__HIP_PLATFORM_HCC__) -#include +#include #else #error("Must define exactly one of __HIP_PLATFORM_HCC__ or __HIP_PLATFORM_NVCC__"); #endif diff --git a/projects/hip/include/hip/nvcc_detail/hip_cooperative_groups.h b/projects/hip/include/hip/nvcc_detail/hip_cooperative_groups.h new file mode 100644 index 0000000000..113e600eec --- /dev/null +++ b/projects/hip/include/hip/nvcc_detail/hip_cooperative_groups.h @@ -0,0 +1,12 @@ +#ifndef HIP_INCLUDE_HIP_NVCC_DETAIL_HIP_COOPERATIVE_GROUPS_H +#define HIP_INCLUDE_HIP_NVCC_DETAIL_HIP_COOPERATIVE_GROUPS_H + +// Include CUDA headers +#include +#include + +// Include HIP wrapper headers around CUDA +#include +#include + +#endif // HIP_INCLUDE_HIP_NVCC_DETAIL_HIP_COOPERATIVE_GROUPS_H