diff --git a/cub/cub/device/dispatch/kernels/merge_sort.cuh b/cub/cub/device/dispatch/kernels/merge_sort.cuh index a5070e353a7..51cfad7dd1e 100644 --- a/cub/cub/device/dispatch/kernels/merge_sort.cuh +++ b/cub/cub/device/dispatch/kernels/merge_sort.cuh @@ -17,17 +17,7 @@ #include #include -THRUST_NAMESPACE_BEGIN - -namespace cuda_cub::core::detail -{ -// We must forward declare here because make_load_iterator.h pulls in non NVRTC compilable code -template -typename detail::LoadIterator::type _CCCL_DEVICE _CCCL_FORCEINLINE -make_load_iterator(PtxPlan const&, It it); -} // namespace cuda_cub::core::detail - -THRUST_NAMESPACE_END +#include CUB_NAMESPACE_BEGIN diff --git a/thrust/thrust/detail/type_traits/pointer_traits.h b/thrust/thrust/detail/type_traits/pointer_traits.h index baf4c0ca1dd..cb1b91b58f7 100644 --- a/thrust/thrust/detail/type_traits/pointer_traits.h +++ b/thrust/thrust/detail/type_traits/pointer_traits.h @@ -31,8 +31,8 @@ #include #include -#include -#include +#include +#include THRUST_NAMESPACE_BEGIN namespace detail @@ -62,7 +62,7 @@ struct pointer_difference template struct pointer_difference { - using type = std::ptrdiff_t; + using type = ::cuda::std::ptrdiff_t; }; template @@ -117,10 +117,10 @@ template