dedf08166a
Added 863 files from NVIDIA/cccl sparse checkout: - c2h/ (27 files): Catch2 test helpers — generators, validators, runner - nvbench_helper/ (10 files): Benchmark harness utilities - cmake/ (29 files): CMake presets and build helpers - cudax/ (794 files): Experimental CUDA extensions - AGENTS.md: NVIDIA's official AI agent instructions for CCCL - CMakePresets.json: Standardized build configurations - cccl-version.json: Version tracking Also added CCCL_ASSET_MAP.md mapping all 4295 CCCL files to competition value and PRD items. cccl_upstream now covers 100% of competition-critical assets: - 27 tuning headers (SM80/90/100 benchmark data) - 32 dispatch headers (algorithm implementations) - 60 Thrust examples (correctness verification) - 217 CUB Catch2 tests (regression matrix) - 153 CUB benchmarks (parameter space search) - 18 CUB examples (API verification) - 27 test helpers + benchmark harness - 794 cudax experimental extensions
69 lines
2.2 KiB
C++
69 lines
2.2 KiB
C++
// SPDX-FileCopyrightText: Copyright (c) 2011-2024, NVIDIA CORPORATION. All rights reserved.
|
|
// SPDX-License-Identifier: BSD-3
|
|
|
|
#pragma once
|
|
|
|
#include <thrust/detail/vector_base.h>
|
|
|
|
#include <cuda/std/type_traits>
|
|
|
|
#include <memory>
|
|
|
|
#include <catch2/catch_tostring.hpp>
|
|
|
|
#if THRUST_DEVICE_SYSTEM == THRUST_DEVICE_SYSTEM_CUDA
|
|
# include <c2h/checked_allocator.cuh>
|
|
#else
|
|
# include <thrust/device_vector.h>
|
|
# include <thrust/host_vector.h>
|
|
#endif
|
|
|
|
namespace c2h
|
|
{
|
|
#if THRUST_DEVICE_SYSTEM == THRUST_DEVICE_SYSTEM_CUDA
|
|
template <typename T>
|
|
using host_vector = THRUST_NS_QUALIFIER::detail::vector_base<T, c2h::checked_host_allocator<T>>;
|
|
|
|
template <typename T>
|
|
using device_vector = THRUST_NS_QUALIFIER::detail::vector_base<T, c2h::checked_cuda_allocator<T>>;
|
|
#else // THRUST_DEVICE_SYSTEM == THRUST_DEVICE_SYSTEM_CUDA
|
|
using THRUST_NS_QUALIFIER::device_vector;
|
|
using THRUST_NS_QUALIFIER::host_vector;
|
|
#endif // THRUST_DEVICE_SYSTEM == THRUST_DEVICE_SYSTEM_CUDA
|
|
} // namespace c2h
|
|
|
|
// We specialize how Catch2 prints ([signed|unsigned]) char vectors for better readability. Let's print them as numbers
|
|
// instead of characters.
|
|
template <typename T, typename A>
|
|
struct Catch::StringMaker<THRUST_NS_QUALIFIER::detail::vector_base<T, A>,
|
|
::cuda::std::enable_if_t<sizeof(T) == 1 && ::cuda::std::is_fundamental_v<T>>>
|
|
{
|
|
// Copied from `rangeToString` in catch_tostring.hpp
|
|
static auto convert(const THRUST_NS_QUALIFIER::detail::vector_base<T, A>& v) -> std::string
|
|
{
|
|
auto first = v.begin();
|
|
auto last = v.end();
|
|
|
|
ReusableStringStream rss;
|
|
rss << "{ ";
|
|
if (first != last)
|
|
{
|
|
rss << Detail::stringify(static_cast<unsigned>(static_cast<T>(*first)));
|
|
for (++first; first != last; ++first)
|
|
{
|
|
rss << ", " << Detail::stringify(static_cast<unsigned>(static_cast<T>(*first)));
|
|
}
|
|
}
|
|
rss << " }";
|
|
return rss.str();
|
|
}
|
|
};
|
|
|
|
// due to an nvcc bug, the above specialization of StringMaker is ambiguous with one inside Catch2, so let's disable
|
|
// Catch2 range formatting for vector_base with sizeof(T) == 1 entirely
|
|
template <typename T, typename A>
|
|
struct Catch::is_range<THRUST_NS_QUALIFIER::detail::vector_base<T, A>>
|
|
{
|
|
static constexpr bool value = !(sizeof(T) == 1 && ::cuda::std::is_fundamental_v<T>);
|
|
};
|