Spaces:
Runtime error
Runtime error
File size: 3,583 Bytes
28958dc |
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 |
#include <unittest/unittest.h>
#include <thrust/functional.h>
#include <thrust/transform.h>
#include <thrust/iterator/constant_iterator.h>
#include <thrust/detail/allocator/allocator_traits.h>
static const size_t num_samples = 10000;
template<typename Vector, typename U> struct rebind_vector;
template<typename T, typename U, typename Allocator>
struct rebind_vector<thrust::host_vector<T, Allocator>, U>
{
typedef typename thrust::detail::allocator_traits<Allocator> alloc_traits;
typedef typename alloc_traits::template rebind_alloc<U> new_alloc;
typedef thrust::host_vector<U, new_alloc> type;
};
template<typename T, typename U, typename Allocator>
struct rebind_vector<thrust::device_vector<T, Allocator>, U>
{
typedef thrust::device_vector<U,
typename Allocator::template rebind<U>::other> type;
};
#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(BitAnd, &, thrust::bit_and, SmallIntegralTypes);
BINARY_FUNCTIONAL_PLACEHOLDERS_TEST(BitOr, |, thrust::bit_or, SmallIntegralTypes);
BINARY_FUNCTIONAL_PLACEHOLDERS_TEST(BitXor, ^, thrust::bit_xor, SmallIntegralTypes);
template<typename T>
struct bit_negate_reference
{
__host__ __device__ T operator()(const T &x) const
{
return ~x;
}
};
template<typename Vector>
void TestFunctionalPlaceholdersBitNegate(void)
{
typedef typename Vector::value_type T;
typedef typename rebind_vector<Vector,bool>::type bool_vector;
Vector input = unittest::random_samples<T>(num_samples);
bool_vector reference(input.size());
thrust::transform(input.begin(), input.end(), reference.begin(), bit_negate_reference<T>());
using namespace thrust::placeholders;
bool_vector result(input.size());
thrust::transform(input.begin(), input.end(), result.begin(), ~_1);
ASSERT_EQUAL(reference, result);
}
DECLARE_INTEGRAL_VECTOR_UNITTEST(TestFunctionalPlaceholdersBitNegate);
|