Spaces:
Runtime error
Runtime error
File size: 1,244 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 54 55 56 57 58 59 60 61 62 63 |
#include <thrust/detail/config.h>
#if THRUST_CPP_DIALECT >= 2011
#include <unittest/unittest.h>
#include <thrust/detail/tuple_algorithms.h>
#include <thrust/type_traits/integer_sequence.h>
// FIXME: Replace with C++14 style `thrust::square<>` when we have it.
struct custom_square
{
template <typename T>
__host__ __device__
T operator()(T v) const
{
return v * v;
}
};
struct custom_square_inplace
{
template <typename T>
__host__ __device__
void operator()(T& v) const
{
v *= v;
}
};
void test_tuple_subset()
{
auto t0 = std::make_tuple(0, 2, 3.14);
auto t1 = thrust::tuple_subset(t0, thrust::index_sequence<2, 0>{});
ASSERT_EQUAL_QUIET(t1, std::make_tuple(3.14, 0));
}
DECLARE_UNITTEST(test_tuple_subset);
void test_tuple_transform()
{
auto t0 = std::make_tuple(0, 2, 3.14);
auto t1 = thrust::tuple_transform(t0, custom_square{});
ASSERT_EQUAL_QUIET(t1, std::make_tuple(0, 4, 9.8596));
}
DECLARE_UNITTEST(test_tuple_transform);
void test_tuple_for_each()
{
auto t = std::make_tuple(0, 2, 3.14);
thrust::tuple_for_each(t, custom_square_inplace{});
ASSERT_EQUAL_QUIET(t, std::make_tuple(0, 4, 9.8596));
}
DECLARE_UNITTEST(test_tuple_for_each);
#endif // THRUST_CPP_DIALECT >= 2011
|