90 lines
7.0 KiB
Plaintext
90 lines
7.0 KiB
Plaintext
|
#include <thrust/functional.h>
|
||
|
|
#include <thrust/transform.h>
|
||
|
|
|
||
|
|
#include <cuda/iterator>
|
||
|
|
|
||
|
|
#include <unittest/unittest.h>
|
||
|
|
|
||
|
|
_CCCL_DIAG_PUSH
|
||
|
|
_CCCL_DIAG_SUPPRESS_MSVC(4244) // warning C4244: '=': conversion from 'int' to '_Ty', possible loss of data
|
||
|
|
|
||
|
|
#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; \
|
||
|
|
using T = typename Vector::value_type; \
|
||
|
|
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(), cuda::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( \
|
||
|
|
cuda::make_constant_iterator<T>(1, zero), \
|
||
|
|
cuda::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, +, ::cuda::std::plus, ThirtyTwoBitTypes);
|
||
|
|
BINARY_FUNCTIONAL_PLACEHOLDERS_TEST(Minus, -, ::cuda::std::minus, ThirtyTwoBitTypes);
|
||
|
|
BINARY_FUNCTIONAL_PLACEHOLDERS_TEST(Multiplies, *, ::cuda::std::multiplies, ThirtyTwoBitTypes);
|
||
|
|
BINARY_FUNCTIONAL_PLACEHOLDERS_TEST(Divides, /, ::cuda::std::divides, ThirtyTwoBitTypes);
|
||
|
|
BINARY_FUNCTIONAL_PLACEHOLDERS_TEST(Modulus, %, ::cuda::std::modulus, SmallIntegralTypes);
|
||
|
|
|
||
|
|
#define UNARY_FUNCTIONAL_PLACEHOLDERS_TEST(name, reference_operator, functor) \
|
||
|
|
template <typename Vector> \
|
||
|
|
void TestFunctionalPlaceholders##name() \
|
||
|
|
{ \
|
||
|
|
static const size_t num_samples = 10000; \
|
||
|
|
using T = typename Vector::value_type; \
|
||
|
|
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, -, ::cuda::std::negate);
|
||
|
|
|
||
|
|
_CCCL_DIAG_POP
|