diff --git a/cub/cub/util_type.cuh b/cub/cub/util_type.cuh index 5afd2dd23aa..30c228c6334 100644 --- a/cub/cub/util_type.cuh +++ b/cub/cub/util_type.cuh @@ -46,7 +46,6 @@ #include #include -#include #include #include @@ -59,6 +58,12 @@ _CCCL_DIAG_POP # endif // !_CCCL_CUDACC_BELOW_11_8 #endif // _CCCL_HAS_NV_BF16 +#ifdef _CCCL_COMPILER_NVRTC +# include +#else // !defined(_CCCL_COMPILER_NVRTC) +# include +#endif // defined(_CCCL_COMPILER_NVRTC) + CUB_NAMESPACE_BEGIN #ifndef CUB_IS_INT128_ENABLED