CCCL (CUDA C++ Core Libraries) provides: - CUB: device/block/warp-level GPU primitives (reduce, scan, sort, topk) - Thrust: high-level parallel algorithms (transform_reduce, sort, scan) - libcudacxx: CUDA C++ standard library (atomics, barriers, memory) - cudax: experimental features (memory resources, allocators) - Tuning policies: per-SM hardware-specific algorithm parameters Competition optimization vectors mapped to CCCL: - Output TPS (83% weight): warp_reduce, block_reduce, device_topk - Input TPS (14% weight): device_scan, block_load, prefetch - Cache TPS (3% weight): prefix caching strategy patterns - Memory (0.9 util): pooled/cached/buddy allocators Source: https://github.com/NVIDIA/cccl (shallow clone, HEAD only) License: Apache-2.0
545 lines
17 KiB
Plaintext
545 lines
17 KiB
Plaintext
#include <thrust/generate.h>
|
|
#include <thrust/swap.h>
|
|
#include <thrust/tuple.h>
|
|
|
|
#include <unittest/unittest.h>
|
|
|
|
using namespace unittest;
|
|
|
|
template <typename T>
|
|
struct TestTupleConstructor
|
|
{
|
|
void operator()()
|
|
{
|
|
using namespace thrust;
|
|
|
|
host_vector<T> data = random_integers<T>(10);
|
|
|
|
tuple<T> t1(data[0]);
|
|
ASSERT_EQUAL(data[0], get<0>(t1));
|
|
|
|
tuple<T, T> t2(data[0], data[1]);
|
|
ASSERT_EQUAL(data[0], get<0>(t2));
|
|
ASSERT_EQUAL(data[1], get<1>(t2));
|
|
|
|
tuple<T, T, T> t3(data[0], data[1], data[2]);
|
|
ASSERT_EQUAL(data[0], get<0>(t3));
|
|
ASSERT_EQUAL(data[1], get<1>(t3));
|
|
ASSERT_EQUAL(data[2], get<2>(t3));
|
|
|
|
tuple<T, T, T, T> t4(data[0], data[1], data[2], data[3]);
|
|
ASSERT_EQUAL(data[0], get<0>(t4));
|
|
ASSERT_EQUAL(data[1], get<1>(t4));
|
|
ASSERT_EQUAL(data[2], get<2>(t4));
|
|
ASSERT_EQUAL(data[3], get<3>(t4));
|
|
|
|
tuple<T, T, T, T, T> t5(data[0], data[1], data[2], data[3], data[4]);
|
|
ASSERT_EQUAL(data[0], get<0>(t5));
|
|
ASSERT_EQUAL(data[1], get<1>(t5));
|
|
ASSERT_EQUAL(data[2], get<2>(t5));
|
|
ASSERT_EQUAL(data[3], get<3>(t5));
|
|
ASSERT_EQUAL(data[4], get<4>(t5));
|
|
|
|
tuple<T, T, T, T, T, T> t6(data[0], data[1], data[2], data[3], data[4], data[5]);
|
|
ASSERT_EQUAL(data[0], get<0>(t6));
|
|
ASSERT_EQUAL(data[1], get<1>(t6));
|
|
ASSERT_EQUAL(data[2], get<2>(t6));
|
|
ASSERT_EQUAL(data[3], get<3>(t6));
|
|
ASSERT_EQUAL(data[4], get<4>(t6));
|
|
ASSERT_EQUAL(data[5], get<5>(t6));
|
|
|
|
tuple<T, T, T, T, T, T, T> t7(data[0], data[1], data[2], data[3], data[4], data[5], data[6]);
|
|
ASSERT_EQUAL(data[0], get<0>(t7));
|
|
ASSERT_EQUAL(data[1], get<1>(t7));
|
|
ASSERT_EQUAL(data[2], get<2>(t7));
|
|
ASSERT_EQUAL(data[3], get<3>(t7));
|
|
ASSERT_EQUAL(data[4], get<4>(t7));
|
|
ASSERT_EQUAL(data[5], get<5>(t7));
|
|
ASSERT_EQUAL(data[6], get<6>(t7));
|
|
|
|
tuple<T, T, T, T, T, T, T, T> t8(data[0], data[1], data[2], data[3], data[4], data[5], data[6], data[7]);
|
|
ASSERT_EQUAL(data[0], get<0>(t8));
|
|
ASSERT_EQUAL(data[1], get<1>(t8));
|
|
ASSERT_EQUAL(data[2], get<2>(t8));
|
|
ASSERT_EQUAL(data[3], get<3>(t8));
|
|
ASSERT_EQUAL(data[4], get<4>(t8));
|
|
ASSERT_EQUAL(data[5], get<5>(t8));
|
|
ASSERT_EQUAL(data[6], get<6>(t8));
|
|
ASSERT_EQUAL(data[7], get<7>(t8));
|
|
|
|
tuple<T, T, T, T, T, T, T, T, T> t9(data[0], data[1], data[2], data[3], data[4], data[5], data[6], data[7], data[8]);
|
|
ASSERT_EQUAL(data[0], get<0>(t9));
|
|
ASSERT_EQUAL(data[1], get<1>(t9));
|
|
ASSERT_EQUAL(data[2], get<2>(t9));
|
|
ASSERT_EQUAL(data[3], get<3>(t9));
|
|
ASSERT_EQUAL(data[4], get<4>(t9));
|
|
ASSERT_EQUAL(data[5], get<5>(t9));
|
|
ASSERT_EQUAL(data[6], get<6>(t9));
|
|
ASSERT_EQUAL(data[7], get<7>(t9));
|
|
ASSERT_EQUAL(data[8], get<8>(t9));
|
|
|
|
tuple<T, T, T, T, T, T, T, T, T, T> t10(
|
|
data[0], data[1], data[2], data[3], data[4], data[5], data[6], data[7], data[8], data[9]);
|
|
ASSERT_EQUAL(data[0], get<0>(t10));
|
|
ASSERT_EQUAL(data[1], get<1>(t10));
|
|
ASSERT_EQUAL(data[2], get<2>(t10));
|
|
ASSERT_EQUAL(data[3], get<3>(t10));
|
|
ASSERT_EQUAL(data[4], get<4>(t10));
|
|
ASSERT_EQUAL(data[5], get<5>(t10));
|
|
ASSERT_EQUAL(data[6], get<6>(t10));
|
|
ASSERT_EQUAL(data[7], get<7>(t10));
|
|
ASSERT_EQUAL(data[8], get<8>(t10));
|
|
ASSERT_EQUAL(data[9], get<9>(t10));
|
|
}
|
|
};
|
|
SimpleUnitTest<TestTupleConstructor, BuiltinNumericTypes> TestTupleConstructorInstance;
|
|
|
|
template <typename T>
|
|
struct TestMakeTuple
|
|
{
|
|
void operator()()
|
|
{
|
|
using namespace thrust;
|
|
|
|
host_vector<T> data = random_integers<T>(10);
|
|
|
|
tuple<T> t1 = make_tuple(data[0]);
|
|
ASSERT_EQUAL(data[0], get<0>(t1));
|
|
|
|
tuple<T, T> t2 = make_tuple(data[0], data[1]);
|
|
ASSERT_EQUAL(data[0], get<0>(t2));
|
|
ASSERT_EQUAL(data[1], get<1>(t2));
|
|
|
|
tuple<T, T, T> t3 = make_tuple(data[0], data[1], data[2]);
|
|
ASSERT_EQUAL(data[0], get<0>(t3));
|
|
ASSERT_EQUAL(data[1], get<1>(t3));
|
|
ASSERT_EQUAL(data[2], get<2>(t3));
|
|
|
|
tuple<T, T, T, T> t4 = make_tuple(data[0], data[1], data[2], data[3]);
|
|
ASSERT_EQUAL(data[0], get<0>(t4));
|
|
ASSERT_EQUAL(data[1], get<1>(t4));
|
|
ASSERT_EQUAL(data[2], get<2>(t4));
|
|
ASSERT_EQUAL(data[3], get<3>(t4));
|
|
|
|
tuple<T, T, T, T, T> t5 = make_tuple(data[0], data[1], data[2], data[3], data[4]);
|
|
ASSERT_EQUAL(data[0], get<0>(t5));
|
|
ASSERT_EQUAL(data[1], get<1>(t5));
|
|
ASSERT_EQUAL(data[2], get<2>(t5));
|
|
ASSERT_EQUAL(data[3], get<3>(t5));
|
|
ASSERT_EQUAL(data[4], get<4>(t5));
|
|
|
|
tuple<T, T, T, T, T, T> t6 = make_tuple(data[0], data[1], data[2], data[3], data[4], data[5]);
|
|
ASSERT_EQUAL(data[0], get<0>(t6));
|
|
ASSERT_EQUAL(data[1], get<1>(t6));
|
|
ASSERT_EQUAL(data[2], get<2>(t6));
|
|
ASSERT_EQUAL(data[3], get<3>(t6));
|
|
ASSERT_EQUAL(data[4], get<4>(t6));
|
|
ASSERT_EQUAL(data[5], get<5>(t6));
|
|
|
|
tuple<T, T, T, T, T, T, T> t7 = make_tuple(data[0], data[1], data[2], data[3], data[4], data[5], data[6]);
|
|
ASSERT_EQUAL(data[0], get<0>(t7));
|
|
ASSERT_EQUAL(data[1], get<1>(t7));
|
|
ASSERT_EQUAL(data[2], get<2>(t7));
|
|
ASSERT_EQUAL(data[3], get<3>(t7));
|
|
ASSERT_EQUAL(data[4], get<4>(t7));
|
|
ASSERT_EQUAL(data[5], get<5>(t7));
|
|
ASSERT_EQUAL(data[6], get<6>(t7));
|
|
|
|
tuple<T, T, T, T, T, T, T, T> t8 =
|
|
make_tuple(data[0], data[1], data[2], data[3], data[4], data[5], data[6], data[7]);
|
|
ASSERT_EQUAL(data[0], get<0>(t8));
|
|
ASSERT_EQUAL(data[1], get<1>(t8));
|
|
ASSERT_EQUAL(data[2], get<2>(t8));
|
|
ASSERT_EQUAL(data[3], get<3>(t8));
|
|
ASSERT_EQUAL(data[4], get<4>(t8));
|
|
ASSERT_EQUAL(data[5], get<5>(t8));
|
|
ASSERT_EQUAL(data[6], get<6>(t8));
|
|
ASSERT_EQUAL(data[7], get<7>(t8));
|
|
|
|
tuple<T, T, T, T, T, T, T, T, T> t9 =
|
|
make_tuple(data[0], data[1], data[2], data[3], data[4], data[5], data[6], data[7], data[8]);
|
|
ASSERT_EQUAL(data[0], get<0>(t9));
|
|
ASSERT_EQUAL(data[1], get<1>(t9));
|
|
ASSERT_EQUAL(data[2], get<2>(t9));
|
|
ASSERT_EQUAL(data[3], get<3>(t9));
|
|
ASSERT_EQUAL(data[4], get<4>(t9));
|
|
ASSERT_EQUAL(data[5], get<5>(t9));
|
|
ASSERT_EQUAL(data[6], get<6>(t9));
|
|
ASSERT_EQUAL(data[7], get<7>(t9));
|
|
ASSERT_EQUAL(data[8], get<8>(t9));
|
|
|
|
tuple<T, T, T, T, T, T, T, T, T, T> t10 =
|
|
make_tuple(data[0], data[1], data[2], data[3], data[4], data[5], data[6], data[7], data[8], data[9]);
|
|
ASSERT_EQUAL(data[0], get<0>(t10));
|
|
ASSERT_EQUAL(data[1], get<1>(t10));
|
|
ASSERT_EQUAL(data[2], get<2>(t10));
|
|
ASSERT_EQUAL(data[3], get<3>(t10));
|
|
ASSERT_EQUAL(data[4], get<4>(t10));
|
|
ASSERT_EQUAL(data[5], get<5>(t10));
|
|
ASSERT_EQUAL(data[6], get<6>(t10));
|
|
ASSERT_EQUAL(data[7], get<7>(t10));
|
|
ASSERT_EQUAL(data[8], get<8>(t10));
|
|
ASSERT_EQUAL(data[9], get<9>(t10));
|
|
}
|
|
};
|
|
SimpleUnitTest<TestMakeTuple, BuiltinNumericTypes> TestMakeTupleInstance;
|
|
|
|
template <typename T>
|
|
struct TestTupleGet
|
|
{
|
|
void operator()()
|
|
{
|
|
using namespace thrust;
|
|
host_vector<T> data = random_integers<T>(10);
|
|
|
|
tuple<T> t1(data[0]);
|
|
ASSERT_EQUAL(data[0], thrust::get<0>(t1));
|
|
|
|
tuple<T, T> t2(data[0], data[1]);
|
|
ASSERT_EQUAL(data[0], thrust::get<0>(t2));
|
|
ASSERT_EQUAL(data[1], thrust::get<1>(t2));
|
|
|
|
tuple<T, T, T> t3 = make_tuple(data[0], data[1], data[2]);
|
|
ASSERT_EQUAL(data[0], thrust::get<0>(t3));
|
|
ASSERT_EQUAL(data[1], thrust::get<1>(t3));
|
|
ASSERT_EQUAL(data[2], thrust::get<2>(t3));
|
|
|
|
tuple<T, T, T, T> t4 = make_tuple(data[0], data[1], data[2], data[3]);
|
|
ASSERT_EQUAL(data[0], thrust::get<0>(t4));
|
|
ASSERT_EQUAL(data[1], thrust::get<1>(t4));
|
|
ASSERT_EQUAL(data[2], thrust::get<2>(t4));
|
|
ASSERT_EQUAL(data[3], thrust::get<3>(t4));
|
|
|
|
tuple<T, T, T, T, T> t5 = make_tuple(data[0], data[1], data[2], data[3], data[4]);
|
|
ASSERT_EQUAL(data[0], thrust::get<0>(t5));
|
|
ASSERT_EQUAL(data[1], thrust::get<1>(t5));
|
|
ASSERT_EQUAL(data[2], thrust::get<2>(t5));
|
|
ASSERT_EQUAL(data[3], thrust::get<3>(t5));
|
|
ASSERT_EQUAL(data[4], thrust::get<4>(t5));
|
|
|
|
tuple<T, T, T, T, T, T> t6 = make_tuple(data[0], data[1], data[2], data[3], data[4], data[5]);
|
|
ASSERT_EQUAL(data[0], thrust::get<0>(t6));
|
|
ASSERT_EQUAL(data[1], thrust::get<1>(t6));
|
|
ASSERT_EQUAL(data[2], thrust::get<2>(t6));
|
|
ASSERT_EQUAL(data[3], thrust::get<3>(t6));
|
|
ASSERT_EQUAL(data[4], thrust::get<4>(t6));
|
|
ASSERT_EQUAL(data[5], thrust::get<5>(t6));
|
|
|
|
tuple<T, T, T, T, T, T, T> t7 = make_tuple(data[0], data[1], data[2], data[3], data[4], data[5], data[6]);
|
|
ASSERT_EQUAL(data[0], thrust::get<0>(t7));
|
|
ASSERT_EQUAL(data[1], thrust::get<1>(t7));
|
|
ASSERT_EQUAL(data[2], thrust::get<2>(t7));
|
|
ASSERT_EQUAL(data[3], thrust::get<3>(t7));
|
|
ASSERT_EQUAL(data[4], thrust::get<4>(t7));
|
|
ASSERT_EQUAL(data[5], thrust::get<5>(t7));
|
|
ASSERT_EQUAL(data[6], thrust::get<6>(t7));
|
|
|
|
tuple<T, T, T, T, T, T, T, T> t8 =
|
|
make_tuple(data[0], data[1], data[2], data[3], data[4], data[5], data[6], data[7]);
|
|
ASSERT_EQUAL(data[0], thrust::get<0>(t8));
|
|
ASSERT_EQUAL(data[1], thrust::get<1>(t8));
|
|
ASSERT_EQUAL(data[2], thrust::get<2>(t8));
|
|
ASSERT_EQUAL(data[3], thrust::get<3>(t8));
|
|
ASSERT_EQUAL(data[4], thrust::get<4>(t8));
|
|
ASSERT_EQUAL(data[5], thrust::get<5>(t8));
|
|
ASSERT_EQUAL(data[6], thrust::get<6>(t8));
|
|
ASSERT_EQUAL(data[7], thrust::get<7>(t8));
|
|
|
|
tuple<T, T, T, T, T, T, T, T, T> t9 =
|
|
make_tuple(data[0], data[1], data[2], data[3], data[4], data[5], data[6], data[7], data[8]);
|
|
ASSERT_EQUAL(data[0], thrust::get<0>(t9));
|
|
ASSERT_EQUAL(data[1], thrust::get<1>(t9));
|
|
ASSERT_EQUAL(data[2], thrust::get<2>(t9));
|
|
ASSERT_EQUAL(data[3], thrust::get<3>(t9));
|
|
ASSERT_EQUAL(data[4], thrust::get<4>(t9));
|
|
ASSERT_EQUAL(data[5], thrust::get<5>(t9));
|
|
ASSERT_EQUAL(data[6], thrust::get<6>(t9));
|
|
ASSERT_EQUAL(data[7], thrust::get<7>(t9));
|
|
ASSERT_EQUAL(data[8], thrust::get<8>(t9));
|
|
|
|
tuple<T, T, T, T, T, T, T, T, T, T> t10 =
|
|
make_tuple(data[0], data[1], data[2], data[3], data[4], data[5], data[6], data[7], data[8], data[9]);
|
|
ASSERT_EQUAL(data[0], thrust::get<0>(t10));
|
|
ASSERT_EQUAL(data[1], thrust::get<1>(t10));
|
|
ASSERT_EQUAL(data[2], thrust::get<2>(t10));
|
|
ASSERT_EQUAL(data[3], thrust::get<3>(t10));
|
|
ASSERT_EQUAL(data[4], thrust::get<4>(t10));
|
|
ASSERT_EQUAL(data[5], thrust::get<5>(t10));
|
|
ASSERT_EQUAL(data[6], thrust::get<6>(t10));
|
|
ASSERT_EQUAL(data[7], thrust::get<7>(t10));
|
|
ASSERT_EQUAL(data[8], thrust::get<8>(t10));
|
|
ASSERT_EQUAL(data[9], thrust::get<9>(t10));
|
|
}
|
|
};
|
|
SimpleUnitTest<TestTupleGet, BuiltinNumericTypes> TestTupleGetInstance;
|
|
|
|
template <typename T>
|
|
struct TestTupleComparison
|
|
{
|
|
void operator()()
|
|
{
|
|
using namespace thrust;
|
|
|
|
tuple<T, T, T, T, T> lhs(0, 0, 0, 0, 0), rhs(0, 0, 0, 0, 0);
|
|
|
|
// equality
|
|
ASSERT_EQUAL(true, lhs == rhs);
|
|
get<0>(rhs) = 1;
|
|
ASSERT_EQUAL(false, lhs == rhs);
|
|
|
|
// inequality
|
|
ASSERT_EQUAL(true, lhs != rhs);
|
|
lhs = rhs;
|
|
ASSERT_EQUAL(false, lhs != rhs);
|
|
|
|
// less than
|
|
lhs = make_tuple(0, 0, 0, 0, 0);
|
|
rhs = make_tuple(0, 0, 1, 0, 0);
|
|
ASSERT_EQUAL(true, lhs < rhs);
|
|
get<0>(lhs) = 2;
|
|
ASSERT_EQUAL(false, lhs < rhs);
|
|
|
|
// less than equal
|
|
lhs = make_tuple(0, 0, 0, 0, 0);
|
|
rhs = lhs;
|
|
ASSERT_EQUAL(true, lhs <= rhs); // equal
|
|
get<2>(rhs) = 1;
|
|
ASSERT_EQUAL(true, lhs <= rhs); // less than
|
|
get<2>(lhs) = 2;
|
|
ASSERT_EQUAL(false, lhs <= rhs);
|
|
|
|
// greater than
|
|
lhs = make_tuple(1, 0, 0, 0, 0);
|
|
rhs = make_tuple(0, 1, 1, 1, 1);
|
|
ASSERT_EQUAL(true, lhs > rhs);
|
|
get<0>(rhs) = 2;
|
|
ASSERT_EQUAL(false, lhs > rhs);
|
|
|
|
// greater than equal
|
|
lhs = make_tuple(0, 0, 0, 0, 0);
|
|
rhs = lhs;
|
|
ASSERT_EQUAL(true, lhs >= rhs); // equal
|
|
get<4>(lhs) = 1;
|
|
ASSERT_EQUAL(true, lhs >= rhs); // greater than
|
|
get<3>(rhs) = 1;
|
|
ASSERT_EQUAL(false, lhs >= rhs);
|
|
}
|
|
};
|
|
SimpleUnitTest<TestTupleComparison, NumericTypes> TestTupleComparisonInstance;
|
|
|
|
template <typename T>
|
|
struct TestTupleTieFunctor
|
|
{
|
|
_CCCL_HOST_DEVICE void clear(T* data) const
|
|
{
|
|
for (int i = 0; i < 10; ++i)
|
|
{
|
|
data[i] = 13;
|
|
}
|
|
}
|
|
|
|
_CCCL_HOST_DEVICE bool operator()() const
|
|
{
|
|
using namespace thrust;
|
|
|
|
bool result = true;
|
|
|
|
T data[10];
|
|
clear(data);
|
|
|
|
// 17 and not 0 to avoid triggering custom_numeric's `operator void *` and a comparison with a null pointer
|
|
// TODO: get this back from 17 to 0 once C++11 is on everywhere and that operator on custom_numeric is changed
|
|
// to an explicit operator bool
|
|
tie(data[0]) = make_tuple(17);
|
|
result &= data[0] == 17;
|
|
clear(data);
|
|
|
|
tie(data[0], data[1]) = make_tuple(17, 1);
|
|
result &= data[0] == 17;
|
|
result &= data[1] == 1;
|
|
clear(data);
|
|
|
|
tie(data[0], data[1], data[2]) = make_tuple(17, 1, 2);
|
|
result &= data[0] == 17;
|
|
result &= data[1] == 1;
|
|
result &= data[2] == 2;
|
|
clear(data);
|
|
|
|
tie(data[0], data[1], data[2], data[3]) = make_tuple(17, 1, 2, 3);
|
|
result &= data[0] == 17;
|
|
result &= data[1] == 1;
|
|
result &= data[2] == 2;
|
|
result &= data[3] == 3;
|
|
clear(data);
|
|
|
|
tie(data[0], data[1], data[2], data[3], data[4]) = make_tuple(17, 1, 2, 3, 4);
|
|
result &= data[0] == 17;
|
|
result &= data[1] == 1;
|
|
result &= data[2] == 2;
|
|
result &= data[3] == 3;
|
|
result &= data[4] == 4;
|
|
clear(data);
|
|
|
|
tie(data[0], data[1], data[2], data[3], data[4], data[5]) = make_tuple(17, 1, 2, 3, 4, 5);
|
|
result &= data[0] == 17;
|
|
result &= data[1] == 1;
|
|
result &= data[2] == 2;
|
|
result &= data[3] == 3;
|
|
result &= data[4] == 4;
|
|
result &= data[5] == 5;
|
|
clear(data);
|
|
|
|
tie(data[0], data[1], data[2], data[3], data[4], data[5], data[6]) = make_tuple(17, 1, 2, 3, 4, 5, 6);
|
|
result &= data[0] == 17;
|
|
result &= data[1] == 1;
|
|
result &= data[2] == 2;
|
|
result &= data[3] == 3;
|
|
result &= data[4] == 4;
|
|
result &= data[5] == 5;
|
|
result &= data[6] == 6;
|
|
clear(data);
|
|
|
|
tie(data[0], data[1], data[2], data[3], data[4], data[5], data[6], data[7]) = make_tuple(17, 1, 2, 3, 4, 5, 6, 7);
|
|
result &= data[0] == 17;
|
|
result &= data[1] == 1;
|
|
result &= data[2] == 2;
|
|
result &= data[3] == 3;
|
|
result &= data[4] == 4;
|
|
result &= data[5] == 5;
|
|
result &= data[6] == 6;
|
|
result &= data[7] == 7;
|
|
clear(data);
|
|
|
|
tie(data[0], data[1], data[2], data[3], data[4], data[5], data[6], data[7], data[8]) =
|
|
make_tuple(17, 1, 2, 3, 4, 5, 6, 7, 8);
|
|
result &= data[0] == 17;
|
|
result &= data[1] == 1;
|
|
result &= data[2] == 2;
|
|
result &= data[3] == 3;
|
|
result &= data[4] == 4;
|
|
result &= data[5] == 5;
|
|
result &= data[6] == 6;
|
|
result &= data[7] == 7;
|
|
result &= data[8] == 8;
|
|
clear(data);
|
|
|
|
tie(data[0], data[1], data[2], data[3], data[4], data[5], data[6], data[7], data[8], data[9]) =
|
|
make_tuple(17, 1, 2, 3, 4, 5, 6, 7, 8, 9);
|
|
result &= data[0] == 17;
|
|
result &= data[1] == 1;
|
|
result &= data[2] == 2;
|
|
result &= data[3] == 3;
|
|
result &= data[4] == 4;
|
|
result &= data[5] == 5;
|
|
result &= data[6] == 6;
|
|
result &= data[7] == 7;
|
|
result &= data[8] == 8;
|
|
result &= data[9] == 9;
|
|
clear(data);
|
|
|
|
return result;
|
|
}
|
|
};
|
|
|
|
template <typename T>
|
|
struct TestTupleTie
|
|
{
|
|
void operator()()
|
|
{
|
|
thrust::host_vector<bool> h_result(1);
|
|
thrust::generate(h_result.begin(), h_result.end(), TestTupleTieFunctor<T>());
|
|
|
|
thrust::device_vector<bool> d_result(1);
|
|
thrust::generate(d_result.begin(), d_result.end(), TestTupleTieFunctor<T>());
|
|
|
|
ASSERT_EQUAL(true, h_result[0]);
|
|
ASSERT_EQUAL(true, d_result[0]);
|
|
}
|
|
};
|
|
SimpleUnitTest<TestTupleTie, NumericTypes> TestTupleTieInstance;
|
|
|
|
void TestTupleSwap()
|
|
{
|
|
int a = 7;
|
|
int b = 13;
|
|
int c = 42;
|
|
|
|
int x = 77;
|
|
int y = 1313;
|
|
int z = 4242;
|
|
|
|
thrust::tuple<int, int, int> t1(a, b, c);
|
|
thrust::tuple<int, int, int> t2(x, y, z);
|
|
|
|
using ::cuda::std::swap;
|
|
swap(t1, t2);
|
|
|
|
ASSERT_EQUAL(x, thrust::get<0>(t1));
|
|
ASSERT_EQUAL(y, thrust::get<1>(t1));
|
|
ASSERT_EQUAL(z, thrust::get<2>(t1));
|
|
ASSERT_EQUAL(a, thrust::get<0>(t2));
|
|
ASSERT_EQUAL(b, thrust::get<1>(t2));
|
|
ASSERT_EQUAL(c, thrust::get<2>(t2));
|
|
|
|
using swappable_tuple = thrust::tuple<user_swappable, user_swappable, user_swappable, user_swappable>;
|
|
|
|
thrust::host_vector<swappable_tuple> h_v1(1), h_v2(1);
|
|
thrust::device_vector<swappable_tuple> d_v1(1), d_v2(1);
|
|
|
|
thrust::swap_ranges(h_v1.begin(), h_v1.end(), h_v2.begin());
|
|
thrust::swap_ranges(d_v1.begin(), d_v1.end(), d_v2.begin());
|
|
|
|
swappable_tuple ref(user_swappable(true), user_swappable(true), user_swappable(true), user_swappable(true));
|
|
|
|
ASSERT_EQUAL_QUIET(ref, h_v1[0]);
|
|
ASSERT_EQUAL_QUIET(ref, h_v1[0]);
|
|
ASSERT_EQUAL_QUIET(ref, (swappable_tuple) d_v1[0]);
|
|
ASSERT_EQUAL_QUIET(ref, (swappable_tuple) d_v1[0]);
|
|
}
|
|
DECLARE_UNITTEST(TestTupleSwap);
|
|
|
|
void TestTupleStructuredBindings()
|
|
{
|
|
const int a = 0;
|
|
const int b = 42;
|
|
const int c = 1337;
|
|
thrust::tuple<int, int, int> t(a, b, c);
|
|
|
|
auto [a2, b2, c2] = t;
|
|
ASSERT_EQUAL(a, a2);
|
|
ASSERT_EQUAL(b, b2);
|
|
ASSERT_EQUAL(c, c2);
|
|
}
|
|
DECLARE_UNITTEST(TestTupleStructuredBindings);
|
|
|
|
void TestTupleCTAD()
|
|
{
|
|
const int a = 0;
|
|
const char b = 42;
|
|
const short c = 1337;
|
|
thrust::tuple t(a, b, c);
|
|
|
|
auto [a2, b2, c2] = t;
|
|
ASSERT_EQUAL(a, a2);
|
|
ASSERT_EQUAL(b, b2);
|
|
ASSERT_EQUAL(c, c2);
|
|
}
|
|
DECLARE_UNITTEST(TestTupleCTAD);
|
|
|
|
void TestTupleOfIteratorReferenceAssignsFromConst()
|
|
{
|
|
// tuple of mutable references
|
|
thrust::device_vector<int> v(10);
|
|
using devref = decltype(v[0]);
|
|
auto refs = thrust::detail::tuple_of_iterator_references<devref>{thrust::tuple<devref>(v[0])};
|
|
|
|
// tuple of const references
|
|
const thrust::device_vector<int> cv(10);
|
|
using devcref = decltype(cv[0]);
|
|
auto crefs = thrust::detail::tuple_of_iterator_references<devcref>{thrust::tuple<devcref>(cv[0])};
|
|
|
|
// should compile:
|
|
refs = crefs;
|
|
}
|
|
DECLARE_UNITTEST(TestTupleOfIteratorReferenceAssignsFromConst);
|