Spaces:
Runtime error
Runtime error
File size: 2,052 Bytes
be11144 |
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 |
#include <unittest/unittest.h>
#include <thrust/functional.h>
#include <thrust/transform.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, reference_operator, functor) \
template<typename Vector> \
void TestFunctionalPlaceholdersBinary##name(void) \
{ \
typedef typename Vector::value_type T; \
typedef typename rebind_vector<Vector,bool>::type bool_vector; \
Vector lhs = unittest::random_samples<T>(num_samples); \
Vector rhs = unittest::random_samples<T>(num_samples); \
\
bool_vector reference(lhs.size()); \
thrust::transform(lhs.begin(), lhs.end(), rhs.begin(), reference.begin(), functor<T>()); \
\
using namespace thrust::placeholders; \
bool_vector result(lhs.size()); \
thrust::transform(lhs.begin(), lhs.end(), rhs.begin(), result.begin(), _1 reference_operator _2); \
\
ASSERT_EQUAL(reference, result); \
} \
DECLARE_VECTOR_UNITTEST(TestFunctionalPlaceholdersBinary##name);
BINARY_FUNCTIONAL_PLACEHOLDERS_TEST(EqualTo, ==, thrust::equal_to);
BINARY_FUNCTIONAL_PLACEHOLDERS_TEST(NotEqualTo, !=, thrust::not_equal_to);
BINARY_FUNCTIONAL_PLACEHOLDERS_TEST(Greater, >, thrust::greater);
BINARY_FUNCTIONAL_PLACEHOLDERS_TEST(Less, <, thrust::less);
BINARY_FUNCTIONAL_PLACEHOLDERS_TEST(GreaterEqual, >=, thrust::greater_equal);
BINARY_FUNCTIONAL_PLACEHOLDERS_TEST(LessEqual, <=, thrust::less_equal);
|