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
|
#include <thrust/functional.h>
#include <thrust/iterator/constant_iterator.h>
#include <thrust/transform.h>
#include <unittest/unittest.h>
#define BINARY_FUNCTIONAL_PLACEHOLDERS_TEST(name, op, reference_functor, type_list) \
template <typename Vector> \
struct TestFunctionalPlaceholders##name \
{ \
void operator()(const size_t) \
{ \
static const size_t num_samples = 10000; \
const size_t zero = 0; \
typedef typename Vector::value_type T; \
Vector lhs = unittest::random_samples<T>(num_samples); \
Vector rhs = unittest::random_samples<T>(num_samples); \
thrust::replace(rhs.begin(), rhs.end(), T(0), T(1)); \
\
Vector reference(lhs.size()); \
Vector result(lhs.size()); \
using namespace thrust::placeholders; \
\
thrust::transform(lhs.begin(), lhs.end(), rhs.begin(), reference.begin(), reference_functor<T>()); \
thrust::transform(lhs.begin(), lhs.end(), rhs.begin(), result.begin(), _1 op _2); \
ASSERT_ALMOST_EQUAL(reference, result); \
\
thrust::transform( \
lhs.begin(), lhs.end(), thrust::make_constant_iterator<T>(1), reference.begin(), reference_functor<T>()); \
thrust::transform(lhs.begin(), lhs.end(), result.begin(), _1 op T(1)); \
ASSERT_ALMOST_EQUAL(reference, result); \
\
thrust::transform( \
thrust::make_constant_iterator<T>(1, zero), \
thrust::make_constant_iterator<T>(1, num_samples), \
rhs.begin(), \
reference.begin(), \
reference_functor<T>()); \
thrust::transform(rhs.begin(), rhs.end(), result.begin(), T(1) op _1); \
ASSERT_ALMOST_EQUAL(reference, result); \
} \
}; \
VectorUnitTest<TestFunctionalPlaceholders##name, type_list, thrust::device_vector, thrust::device_allocator> \
TestFunctionalPlaceholders##name##DeviceInstance; \
VectorUnitTest<TestFunctionalPlaceholders##name, type_list, thrust::host_vector, std::allocator> \
TestFunctionalPlaceholders##name##HostInstance;
BINARY_FUNCTIONAL_PLACEHOLDERS_TEST(Plus, +, thrust::plus, ThirtyTwoBitTypes);
BINARY_FUNCTIONAL_PLACEHOLDERS_TEST(Minus, -, thrust::minus, ThirtyTwoBitTypes);
BINARY_FUNCTIONAL_PLACEHOLDERS_TEST(Multiplies, *, thrust::multiplies, ThirtyTwoBitTypes);
BINARY_FUNCTIONAL_PLACEHOLDERS_TEST(Divides, /, thrust::divides, ThirtyTwoBitTypes);
BINARY_FUNCTIONAL_PLACEHOLDERS_TEST(Modulus, %, thrust::modulus, SmallIntegralTypes);
#define UNARY_FUNCTIONAL_PLACEHOLDERS_TEST(name, reference_operator, functor) \
template <typename Vector> \
void TestFunctionalPlaceholders##name() \
{ \
static const size_t num_samples = 10000; \
typedef typename Vector::value_type T; \
Vector input = unittest::random_samples<T>(num_samples); \
\
Vector reference(input.size()); \
thrust::transform(input.begin(), input.end(), reference.begin(), functor<T>()); \
\
using namespace thrust::placeholders; \
Vector result(input.size()); \
thrust::transform(input.begin(), input.end(), result.begin(), reference_operator _1); \
\
ASSERT_EQUAL(reference, result); \
} \
DECLARE_VECTOR_UNITTEST(TestFunctionalPlaceholders##name);
template <typename T>
struct unary_plus_reference
{
_CCCL_HOST_DEVICE T operator()(const T& x) const
{ // Static cast to undo integral promotion
return static_cast<T>(+x);
}
};
UNARY_FUNCTIONAL_PLACEHOLDERS_TEST(UnaryPlus, +, unary_plus_reference);
UNARY_FUNCTIONAL_PLACEHOLDERS_TEST(Negate, -, thrust::negate);
|
