58 lines
3.3 KiB
Plaintext
58 lines
3.3 KiB
Plaintext
|
#include <thrust/detail/allocator/allocator_system.h>
|
||
|
|
#include <thrust/functional.h>
|
||
|
|
#include <thrust/transform.h>
|
||
|
|
|
||
|
|
#include <unittest/unittest.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>
|
||
|
|
{
|
||
|
|
using alloc_traits = typename cuda::std::allocator_traits<Allocator>;
|
||
|
|
using new_alloc = typename alloc_traits::template rebind_alloc<U>;
|
||
|
|
using type = thrust::host_vector<U, new_alloc>;
|
||
|
|
};
|
||
|
|
|
||
|
|
template <typename T, typename U, typename Allocator>
|
||
|
|
struct rebind_vector<thrust::device_vector<T, Allocator>, U>
|
||
|
|
{
|
||
|
|
using type = thrust::device_vector<U, typename Allocator::template rebind<U>::other>;
|
||
|
|
};
|
||
|
|
|
||
|
|
template <typename T, typename U, typename Allocator>
|
||
|
|
struct rebind_vector<thrust::universal_vector<T, Allocator>, U>
|
||
|
|
{
|
||
|
|
using type = thrust::universal_vector<U, typename Allocator::template rebind<U>::other>;
|
||
|
|
};
|
||
|
|
|
||
|
|
#define BINARY_FUNCTIONAL_PLACEHOLDERS_TEST(name, reference_operator, functor) \
|
||
|
|
template <typename Vector> \
|
||
|
|
void TestFunctionalPlaceholdersBinary##name() \
|
||
|
|
{ \
|
||
|
|
using T = typename Vector::value_type; \
|
||
|
|
using bool_vector = typename rebind_vector<Vector, bool>::type; \
|
||
|
|
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, ==, ::cuda::std::equal_to);
|
||
|
|
BINARY_FUNCTIONAL_PLACEHOLDERS_TEST(NotEqualTo, !=, ::cuda::std::not_equal_to);
|
||
|
|
BINARY_FUNCTIONAL_PLACEHOLDERS_TEST(Greater, >, ::cuda::std::greater);
|
||
|
|
BINARY_FUNCTIONAL_PLACEHOLDERS_TEST(Less, <, ::cuda::std::less);
|
||
|
|
BINARY_FUNCTIONAL_PLACEHOLDERS_TEST(GreaterEqual, >=, ::cuda::std::greater_equal);
|
||
|
|
BINARY_FUNCTIONAL_PLACEHOLDERS_TEST(LessEqual, <=, ::cuda::std::less_equal);
|