File tree 2 files changed +7
-6
lines changed
2 files changed +7
-6
lines changed Original file line number Diff line number Diff line change 46
46
#include < cub/detail/type_traits.cuh>
47
47
#include < cub/detail/uninitialized_copy.cuh>
48
48
49
- #include < thrust/iterator/discard_iterator .h>
49
+ #include < thrust/iterator/detail/any_assign .h>
50
50
51
51
#include < cuda/std/cstdint>
52
52
#include < cuda/std/iterator>
53
53
#include < cuda/std/limits>
54
- #include < cuda/std/ type_traits>
54
+ #include < cuda/type_traits>
55
55
56
56
#if _CCCL_HAS_NVFP16()
57
57
# include < cuda_fp16.h>
@@ -113,11 +113,11 @@ struct non_void_value_impl
113
113
template <typename It, typename FallbackT>
114
114
struct non_void_value_impl <It, FallbackT, false >
115
115
{
116
- // we consider thrust::discard_iterator's value_type as `void` as well, so users can switch from
116
+ // we consider thrust::discard_iterator's value_type (`any_assign`) as `void` as well, so users can switch from
117
117
// cub::DiscardInputIterator to thrust::discard_iterator.
118
118
using type =
119
119
::cuda::std::_If<::cuda::std::is_void_v<it_value_t <It>>
120
- || ::cuda::std::is_same_v<it_value_t <It>, THRUST_NS_QUALIFIER::discard_iterator<>::value_type >,
120
+ || ::cuda::std::is_same_v<it_value_t <It>, THRUST_NS_QUALIFIER::detail::any_assign >,
121
121
FallbackT,
122
122
it_value_t <It>>;
123
123
};
Original file line number Diff line number Diff line change @@ -164,10 +164,11 @@ class discard_iterator : public detail::make_discard_iterator_base<System>::type
164
164
// ! parameter is \c 0.
165
165
// ! \return A new \p discard_iterator with index as given by \p i.
166
166
// ! \see constant_iterator
167
+ template <typename System = use_default>
167
168
inline _CCCL_HOST_DEVICE discard_iterator<>
168
- make_discard_iterator (discard_iterator<>::difference_type i = discard_iterator<>::difference_type( 0 ) )
169
+ make_discard_iterator (typename discard_iterator<System >::difference_type i = {} )
169
170
{
170
- return discard_iterator<>(i);
171
+ return discard_iterator<System >(i);
171
172
}
172
173
173
174
// ! \} // end fancyiterators
You can’t perform that action at this time.
0 commit comments