File tree Expand file tree Collapse file tree 1 file changed +5
-3
lines changed Expand file tree Collapse file tree 1 file changed +5
-3
lines changed Original file line number Diff line number Diff line change 77#include < cub/device/device_scan.cuh>
88#include < cub/util_allocator.cuh>
99
10+ #include < thrust/iterator/counting_iterator.h>
11+ #include < thrust/iterator/transform_iterator.h>
12+
1013#include < cuda/iterator>
1114#include < cuda/std/type_traits>
1215
@@ -316,11 +319,10 @@ void TestAlgorithmSpecialisation()
316319
317320 using RunItemT = float ;
318321 using RunLengthT = uint32_t ;
319- using CasterOp = CastOp<RunItemT>;
320- using ItemItT = cuda::transform_iterator<CasterOp, cuda::counting_iterator<uint64_t >>;
322+ using ItemItT = thrust::counting_iterator<RunItemT>;
321323 using RunLengthsItT = cuda::transform_iterator<ModOp, cuda::counting_iterator<RunLengthT>>;
322324
323- ItemItT d_unique_items (cuda::counting_iterator< uint64_t >( 1000U ), CasterOp{} );
325+ ItemItT d_unique_items (1000U );
324326 RunLengthsItT d_run_lengths (cuda::counting_iterator<RunLengthT>(0 ), ModOp{});
325327
326328 constexpr uint32_t num_runs = 10000 ;
You can’t perform that action at this time.
0 commit comments