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
78 lines
2.2 KiB
Plaintext
78 lines
2.2 KiB
Plaintext
#define CCCL_IGNORE_DEPRECATED_API
|
|
|
|
// Wrap thrust and cub in different enclosing namespaces
|
|
// (In practice, you probably want these to be the same, in which case just
|
|
// set THRUST_CUB_WRAPPED_NAMESPACE to set both).
|
|
#define THRUST_WRAPPED_NAMESPACE wrap_thrust
|
|
#define CUB_WRAPPED_NAMESPACE wrap_cub
|
|
|
|
// Enable error checking:
|
|
#define CUB_STDERR
|
|
|
|
#include <cub/device/device_radix_sort.cuh>
|
|
#include <cub/util_debug.cuh>
|
|
|
|
#include <thrust/device_vector.h>
|
|
#include <thrust/host_vector.h>
|
|
#include <thrust/sort.h>
|
|
|
|
#include <cstdint>
|
|
#include <cstdlib>
|
|
|
|
#include "test_util.h"
|
|
|
|
// Test that we can use a few common utilities and algorithms from wrapped
|
|
// Thrust/CUB namespaces at runtime. More extensive testing is performed by the
|
|
// header tests and the check_namespace.cmake test.
|
|
int main(int argc, char** argv)
|
|
{
|
|
CommandLineArgs args(argc, argv);
|
|
CubDebugExit(args.DeviceInit());
|
|
|
|
constexpr std::size_t n = 2048;
|
|
|
|
// Fill a vector with random data:
|
|
::wrap_thrust::thrust::host_vector<int> h_input(n);
|
|
for (auto& val : h_input)
|
|
{
|
|
RandomBits(val);
|
|
}
|
|
|
|
// Test the qualifier macro:
|
|
THRUST_NS_QUALIFIER::device_vector<int> d_input(h_input);
|
|
THRUST_NS_QUALIFIER::device_vector<int> d_output(n);
|
|
|
|
std::size_t temp_storage_bytes{};
|
|
|
|
// Sort with DeviceRadixSort:
|
|
auto error = ::wrap_cub::cub::DeviceRadixSort::SortKeys(
|
|
nullptr,
|
|
temp_storage_bytes,
|
|
::wrap_thrust::thrust::raw_pointer_cast(d_input.data()),
|
|
::wrap_thrust::thrust::raw_pointer_cast(d_output.data()),
|
|
static_cast<std::size_t>(n));
|
|
|
|
CubDebugExit(error);
|
|
|
|
::wrap_thrust::thrust::device_vector<std::uint8_t> temp_storage(temp_storage_bytes);
|
|
|
|
// Test the CUB qualifier macro:
|
|
error = CUB_NS_QUALIFIER::DeviceRadixSort::SortKeys(
|
|
::wrap_thrust::thrust::raw_pointer_cast(temp_storage.data()),
|
|
temp_storage_bytes,
|
|
::wrap_thrust::thrust::raw_pointer_cast(d_input.data()),
|
|
::wrap_thrust::thrust::raw_pointer_cast(d_output.data()),
|
|
static_cast<std::size_t>(n));
|
|
|
|
CubDebugExit(error);
|
|
|
|
// Verify output:
|
|
if (!::wrap_thrust::thrust::is_sorted(d_output.cbegin(), d_output.cend()))
|
|
{
|
|
std::cerr << "Output is not sorted!\n";
|
|
return EXIT_FAILURE;
|
|
}
|
|
|
|
return EXIT_SUCCESS;
|
|
}
|