Skip to content

Commit 34a87ff

Browse files
Revert to thrust::counting_iterator
1 parent 5d89b4a commit 34a87ff

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

cub/test/catch2_test_block_run_length_decode.cu

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,9 @@
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;

0 commit comments

Comments
 (0)