1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
#include <thrust/functional.h>
#include <thrust/iterator/retag.h>
#include <thrust/partition.h>
#include <unittest/unittest.h>
template <typename T>
struct is_even
{
_CCCL_HOST_DEVICE bool operator()(T x) const
{
return ((int) x % 2) == 0;
}
};
template <typename Vector>
void TestPartitionPointSimple()
{
typedef typename Vector::value_type T;
typedef typename Vector::iterator Iterator;
Vector v(4);
v[0] = 1;
v[1] = 1;
v[2] = 1;
v[3] = 0;
Iterator first = v.begin();
Iterator last = v.begin() + 4;
Iterator ref = first + 3;
ASSERT_EQUAL_QUIET(ref, thrust::partition_point(first, last, thrust::identity<T>()));
last = v.begin() + 3;
ref = last;
ASSERT_EQUAL_QUIET(ref, thrust::partition_point(first, last, thrust::identity<T>()));
}
DECLARE_VECTOR_UNITTEST(TestPartitionPointSimple);
template <class Vector>
void TestPartitionPoint()
{
typedef typename Vector::value_type T;
typedef typename Vector::iterator Iterator;
const size_t n = (1 << 16) + 13;
Vector v = unittest::random_integers<T>(n);
Iterator ref = thrust::stable_partition(v.begin(), v.end(), is_even<T>());
ASSERT_EQUAL(ref - v.begin(), thrust::partition_point(v.begin(), v.end(), is_even<T>()) - v.begin());
}
DECLARE_INTEGRAL_VECTOR_UNITTEST(TestPartitionPoint);
template <typename ForwardIterator, typename Predicate>
ForwardIterator partition_point(my_system& system, ForwardIterator first, ForwardIterator, Predicate)
{
system.validate_dispatch();
return first;
}
void TestPartitionPointDispatchExplicit()
{
thrust::device_vector<int> vec(1);
my_system sys(0);
thrust::partition_point(sys, vec.begin(), vec.begin(), 0);
ASSERT_EQUAL(true, sys.is_valid());
}
DECLARE_UNITTEST(TestPartitionPointDispatchExplicit);
template <typename ForwardIterator, typename Predicate>
ForwardIterator partition_point(my_tag, ForwardIterator first, ForwardIterator, Predicate)
{
*first = 13;
return first;
}
void TestPartitionPointDispatchImplicit()
{
thrust::device_vector<int> vec(1);
thrust::partition_point(thrust::retag<my_tag>(vec.begin()), thrust::retag<my_tag>(vec.begin()), 0);
ASSERT_EQUAL(13, vec.front());
}
DECLARE_UNITTEST(TestPartitionPointDispatchImplicit);
struct test_less_than
{
long long expected;
_CCCL_DEVICE bool operator()(long long y)
{
return y < expected;
}
};
void TestPartitionPointWithBigIndexesHelper(int magnitude)
{
thrust::counting_iterator<long long> begin(0);
thrust::counting_iterator<long long> end = begin + (1ll << magnitude);
ASSERT_EQUAL(thrust::distance(begin, end), 1ll << magnitude);
test_less_than fn = {(1ll << magnitude) - 17};
ASSERT_EQUAL(thrust::distance(begin, thrust::partition_point(thrust::device, begin, end, fn)),
(1ll << magnitude) - 17);
}
void TestPartitionPointWithBigIndexes()
{
TestPartitionPointWithBigIndexesHelper(30);
TestPartitionPointWithBigIndexesHelper(31);
TestPartitionPointWithBigIndexesHelper(32);
TestPartitionPointWithBigIndexesHelper(33);
}
DECLARE_UNITTEST(TestPartitionPointWithBigIndexes);
|
