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
98 lines
2.5 KiB
C++
98 lines
2.5 KiB
C++
#pragma once
|
|
|
|
#include <thrust/detail/type_traits.h>
|
|
#include <thrust/host_vector.h>
|
|
#include <thrust/random.h>
|
|
|
|
#include <limits>
|
|
|
|
namespace unittest
|
|
{
|
|
inline unsigned int hash(unsigned int a)
|
|
{
|
|
a = (a + 0x7ed55d16) + (a << 12);
|
|
a = (a ^ 0xc761c23c) ^ (a >> 19);
|
|
a = (a + 0x165667b1) + (a << 5);
|
|
a = (a + 0xd3a2646c) ^ (a << 9);
|
|
a = (a + 0xfd7046c5) + (a << 3);
|
|
a = (a ^ 0xb55a4f09) ^ (a >> 16);
|
|
return a;
|
|
}
|
|
|
|
template <typename T>
|
|
struct generate_random_integer
|
|
{
|
|
T operator()(unsigned int i) const
|
|
{
|
|
THRUST_NS_QUALIFIER::default_random_engine rng(hash(i));
|
|
if constexpr (::cuda::std::is_same_v<T, bool>)
|
|
{
|
|
THRUST_NS_QUALIFIER::uniform_int_distribution<unsigned int> dist(0, 1);
|
|
return dist(rng) == 1;
|
|
}
|
|
else if constexpr (::cuda::std::is_integral_v<T>)
|
|
{
|
|
T const min = ::cuda::std::numeric_limits<T>::min();
|
|
T const max = ::cuda::std::numeric_limits<T>::max();
|
|
THRUST_NS_QUALIFIER::uniform_int_distribution<T> dist(min, max);
|
|
return static_cast<T>(dist(rng));
|
|
}
|
|
else if constexpr (::cuda::std::is_floating_point_v<T>)
|
|
{
|
|
T const min = ::cuda::std::numeric_limits<T>::lowest();
|
|
T const max = ::cuda::std::numeric_limits<T>::max();
|
|
THRUST_NS_QUALIFIER::uniform_real_distribution<T> dist(min, max);
|
|
return static_cast<T>(dist(rng));
|
|
}
|
|
else
|
|
{
|
|
return static_cast<T>(rng());
|
|
}
|
|
}
|
|
};
|
|
|
|
template <typename T>
|
|
struct generate_random_sample
|
|
{
|
|
T operator()(unsigned int i) const
|
|
{
|
|
THRUST_NS_QUALIFIER::default_random_engine rng(hash(i));
|
|
THRUST_NS_QUALIFIER::uniform_int_distribution<unsigned int> dist(0, 20);
|
|
|
|
return static_cast<T>(dist(rng));
|
|
}
|
|
};
|
|
|
|
template <typename T>
|
|
THRUST_NS_QUALIFIER::host_vector<T> random_integers(const size_t N)
|
|
{
|
|
THRUST_NS_QUALIFIER::host_vector<T> vec(N);
|
|
THRUST_NS_QUALIFIER::transform(
|
|
THRUST_NS_QUALIFIER::counting_iterator{0u},
|
|
THRUST_NS_QUALIFIER::counting_iterator{static_cast<unsigned int>(N)},
|
|
vec.begin(),
|
|
generate_random_integer<T>());
|
|
|
|
return vec;
|
|
}
|
|
|
|
template <typename T>
|
|
T random_integer()
|
|
{
|
|
return generate_random_integer<T>()(0);
|
|
}
|
|
|
|
template <typename T>
|
|
THRUST_NS_QUALIFIER::host_vector<T> random_samples(const size_t N)
|
|
{
|
|
THRUST_NS_QUALIFIER::host_vector<T> vec(N);
|
|
THRUST_NS_QUALIFIER::transform(
|
|
THRUST_NS_QUALIFIER::counting_iterator{0u},
|
|
THRUST_NS_QUALIFIER::counting_iterator{static_cast<unsigned int>(N)},
|
|
vec.begin(),
|
|
generate_random_sample<T>());
|
|
|
|
return vec;
|
|
}
|
|
}; // end namespace unittest
|