56fd68e7dd
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
134 lines
3.9 KiB
Plaintext
134 lines
3.9 KiB
Plaintext
#include <thrust/iterator/discard_iterator.h>
|
|
|
|
#include <cuda/std/tuple>
|
|
#include <cuda/std/type_traits>
|
|
|
|
#include <unittest/unittest.h>
|
|
|
|
// ensure that we properly support thrust::discard_iterator from cuda::std
|
|
void TestDiscardIteratorTraits()
|
|
{
|
|
using it = thrust::discard_iterator<>;
|
|
using traits = cuda::std::iterator_traits<it>;
|
|
using category = thrust::detail::iterator_category_with_system_and_traversal<::cuda::std::random_access_iterator_tag,
|
|
thrust::any_system_tag,
|
|
thrust::random_access_traversal_tag>;
|
|
|
|
static_assert(cuda::std::is_same_v<traits::difference_type, ptrdiff_t>);
|
|
static_assert(cuda::std::is_same_v<traits::value_type, thrust::detail::any_assign>);
|
|
static_assert(cuda::std::is_same_v<traits::pointer, void>);
|
|
static_assert(cuda::std::is_same_v<traits::reference, thrust::detail::any_assign&>);
|
|
static_assert(cuda::std::is_same_v<traits::iterator_category, category>);
|
|
|
|
static_assert(cuda::std::is_same_v<thrust::iterator_traversal_t<it>, thrust::random_access_traversal_tag>);
|
|
|
|
static_assert(cuda::std::__has_random_access_traversal<it>);
|
|
|
|
static_assert(cuda::std::output_iterator<it, int>);
|
|
static_assert(cuda::std::input_iterator<it>);
|
|
static_assert(cuda::std::forward_iterator<it>);
|
|
static_assert(cuda::std::bidirectional_iterator<it>);
|
|
static_assert(cuda::std::random_access_iterator<it>);
|
|
static_assert(!cuda::std::contiguous_iterator<it>);
|
|
}
|
|
DECLARE_UNITTEST(TestDiscardIteratorTraits);
|
|
|
|
void TestDiscardIteratorIncrement()
|
|
{
|
|
thrust::discard_iterator<> lhs(0);
|
|
thrust::discard_iterator<> rhs(0);
|
|
|
|
ASSERT_EQUAL(0, lhs - rhs);
|
|
|
|
lhs++;
|
|
|
|
ASSERT_EQUAL(1, lhs - rhs);
|
|
|
|
lhs++;
|
|
lhs++;
|
|
|
|
ASSERT_EQUAL(3, lhs - rhs);
|
|
|
|
lhs += 5;
|
|
|
|
ASSERT_EQUAL(8, lhs - rhs);
|
|
|
|
lhs -= 10;
|
|
|
|
ASSERT_EQUAL(-2, lhs - rhs);
|
|
}
|
|
DECLARE_UNITTEST(TestDiscardIteratorIncrement);
|
|
static_assert(cuda::std::is_trivially_copy_constructible<thrust::discard_iterator<>>::value);
|
|
static_assert(cuda::std::is_trivially_copyable<thrust::discard_iterator<>>::value);
|
|
|
|
void TestDiscardIteratorComparison()
|
|
{
|
|
thrust::discard_iterator<> iter1(0);
|
|
thrust::discard_iterator<> iter2(0);
|
|
|
|
ASSERT_EQUAL(0, iter1 - iter2);
|
|
ASSERT_EQUAL(true, iter1 == iter2);
|
|
|
|
iter1++;
|
|
|
|
ASSERT_EQUAL(1, iter1 - iter2);
|
|
ASSERT_EQUAL(false, iter1 == iter2);
|
|
|
|
iter2++;
|
|
|
|
ASSERT_EQUAL(0, iter1 - iter2);
|
|
ASSERT_EQUAL(true, iter1 == iter2);
|
|
|
|
iter1 += 100;
|
|
iter2 += 100;
|
|
|
|
ASSERT_EQUAL(0, iter1 - iter2);
|
|
ASSERT_EQUAL(true, iter1 == iter2);
|
|
}
|
|
DECLARE_UNITTEST(TestDiscardIteratorComparison);
|
|
|
|
void TestMakeDiscardIterator()
|
|
{
|
|
thrust::discard_iterator<> iter0 = thrust::make_discard_iterator(13);
|
|
|
|
*iter0 = 7;
|
|
|
|
thrust::discard_iterator<> iter1 = thrust::make_discard_iterator(7);
|
|
|
|
*iter1 = 13;
|
|
|
|
ASSERT_EQUAL(6, iter0 - iter1);
|
|
}
|
|
DECLARE_UNITTEST(TestMakeDiscardIterator);
|
|
|
|
void TestZippedDiscardIterator()
|
|
{
|
|
using IteratorTuple1 = cuda::std::tuple<thrust::discard_iterator<>>;
|
|
using ZipIterator1 = thrust::zip_iterator<IteratorTuple1>;
|
|
|
|
IteratorTuple1 t = cuda::std::tuple(thrust::make_discard_iterator());
|
|
|
|
ZipIterator1 z_iter1_first = thrust::make_zip_iterator(t);
|
|
ZipIterator1 z_iter1_last = z_iter1_first + 10;
|
|
for (; z_iter1_first != z_iter1_last; ++z_iter1_first)
|
|
{
|
|
;
|
|
}
|
|
|
|
ASSERT_EQUAL(10, cuda::std::get<0>(z_iter1_first.get_iterator_tuple()) - thrust::make_discard_iterator());
|
|
|
|
using IteratorTuple2 = cuda::std::tuple<int*, thrust::discard_iterator<>>;
|
|
using ZipIterator2 = thrust::zip_iterator<IteratorTuple2>;
|
|
|
|
ZipIterator2 z_iter_first = thrust::make_zip_iterator((int*) nullptr, thrust::make_discard_iterator());
|
|
ZipIterator2 z_iter_last = z_iter_first + 10;
|
|
|
|
for (; z_iter_first != z_iter_last; ++z_iter_first)
|
|
{
|
|
;
|
|
}
|
|
|
|
ASSERT_EQUAL(10, cuda::std::get<1>(z_iter_first.get_iterator_tuple()) - thrust::make_discard_iterator());
|
|
}
|
|
DECLARE_UNITTEST(TestZippedDiscardIterator);
|