[CCCL] Add missing CCCL components: c2h, nvbench_helper, cmake, cudax, AGENTS.md
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
This commit is contained in:
muh-bot
289
cccl_upstream/cudax/test/cufile/driver_attributes.cu
Normal file
289
cccl_upstream/cudax/test/cufile/driver_attributes.cu
Normal file
@@ -0,0 +1,289 @@
|
||||
//===----------------------------------------------------------------------===//
|
||||
//
|
||||
// Part of CUDA Experimental in CUDA C++ Core Libraries,
|
||||
// under the Apache License v2.0 with LLVM Exceptions.
|
||||
// See https://llvm.org/LICENSE.txt for license information.
|
||||
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
||||
// SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES.
|
||||
//
|
||||
//===----------------------------------------------------------------------===//
|
||||
|
||||
#include <cuda/std/type_traits>
|
||||
#include <cuda/std/utility>
|
||||
|
||||
#include <cuda/experimental/cufile.cuh>
|
||||
|
||||
#include <stdexcept>
|
||||
|
||||
#include <testing.cuh>
|
||||
|
||||
void test_is_open(bool is_open)
|
||||
{
|
||||
STATIC_REQUIRE(cuda::std::is_same_v<bool, decltype(cudax::cufile_driver.is_open())>);
|
||||
STATIC_REQUIRE(noexcept(cudax::cufile_driver.is_open()));
|
||||
|
||||
REQUIRE(cudax::cufile_driver.is_open() == is_open);
|
||||
}
|
||||
|
||||
template <class Attr>
|
||||
void test_attribute_range([[maybe_unused]] const Attr& attr, [[maybe_unused]] const bool is_open)
|
||||
{
|
||||
#if _CCCL_CTK_AT_LEAST(13, 0)
|
||||
STATIC_REQUIRE(cuda::std::is_same_v<cudax::cufile_driver_attribute_range<Attr>,
|
||||
decltype(cudax::cufile_driver.attribute_range(attr))>);
|
||||
STATIC_REQUIRE(!noexcept(cudax::cufile_driver.attribute_range(attr)));
|
||||
|
||||
const bool expect_success =
|
||||
is_open
|
||||
|| !(cuda::std::is_same_v<Attr, cudax::cufile_driver_attributes::max_device_cache_size_kb_t>
|
||||
|| cuda::std::is_same_v<Attr, cudax::cufile_driver_attributes::max_device_pinned_mem_size_kb_t>);
|
||||
|
||||
if (expect_success)
|
||||
{
|
||||
CHECK_NOTHROW((void) cudax::cufile_driver.attribute_range(attr));
|
||||
}
|
||||
else
|
||||
{
|
||||
CHECK_THROWS_AS((void) cudax::cufile_driver.attribute_range(attr), std::runtime_error);
|
||||
}
|
||||
#endif // _CCCL_CTK_AT_LEAST(13, 0)
|
||||
}
|
||||
|
||||
void test_attribute_ranges(const bool is_open)
|
||||
{
|
||||
using namespace cudax::cufile_driver_attributes;
|
||||
|
||||
// CUFileSizeTConfigParameter_t
|
||||
test_attribute_range(max_io_queue_depth, is_open);
|
||||
test_attribute_range(max_io_threads, is_open);
|
||||
test_attribute_range(min_io_threshold_size_kb, is_open);
|
||||
test_attribute_range(max_request_parallelism, is_open);
|
||||
test_attribute_range(max_direct_io_size_kb, is_open);
|
||||
test_attribute_range(max_device_cache_size_kb, is_open);
|
||||
test_attribute_range(per_buffer_cache_size_kb, is_open);
|
||||
test_attribute_range(max_device_pinned_mem_size_kb, is_open);
|
||||
test_attribute_range(io_batchsize, is_open);
|
||||
test_attribute_range(pollthreshold_size_kb, is_open);
|
||||
test_attribute_range(batch_io_timeout_ms, is_open);
|
||||
|
||||
// CUFileBoolConfigParameter_t are unsupported
|
||||
// CUfileDriverStatusFlags are unsupported
|
||||
// CUfileFeatureFlags are unsupported
|
||||
}
|
||||
|
||||
template <class Attr>
|
||||
void test_get_attribute(const Attr& attr, const bool is_open)
|
||||
{
|
||||
using ValueType = typename Attr::type;
|
||||
|
||||
STATIC_REQUIRE(!noexcept(cudax::cufile_driver.attribute(attr)));
|
||||
STATIC_REQUIRE(cuda::std::is_same_v<ValueType, decltype(cudax::cufile_driver.attribute(attr))>);
|
||||
|
||||
if (is_open || Attr::__can_be_get_when_closed)
|
||||
{
|
||||
CHECK_NOTHROW((void) cudax::cufile_driver.attribute(attr));
|
||||
}
|
||||
else
|
||||
{
|
||||
CHECK_THROWS_AS((void) cudax::cufile_driver.attribute(attr), std::runtime_error);
|
||||
}
|
||||
}
|
||||
|
||||
void test_get_attributes(const bool is_open)
|
||||
{
|
||||
using namespace cudax::cufile_driver_attributes;
|
||||
|
||||
// CUFileSizeTConfigParameter_t
|
||||
test_get_attribute(max_io_queue_depth, is_open);
|
||||
test_get_attribute(max_io_threads, is_open);
|
||||
test_get_attribute(min_io_threshold_size_kb, is_open);
|
||||
test_get_attribute(max_request_parallelism, is_open);
|
||||
test_get_attribute(max_direct_io_size_kb, is_open);
|
||||
test_get_attribute(max_device_cache_size_kb, is_open);
|
||||
test_get_attribute(per_buffer_cache_size_kb, is_open);
|
||||
test_get_attribute(max_device_pinned_mem_size_kb, is_open);
|
||||
test_get_attribute(io_batchsize, is_open);
|
||||
test_get_attribute(pollthreshold_size_kb, is_open);
|
||||
test_get_attribute(batch_io_timeout_ms, is_open);
|
||||
|
||||
// CUFileBoolConfigParameter_t
|
||||
test_get_attribute(use_poll_mode, is_open);
|
||||
test_get_attribute(allow_compat_mode, is_open);
|
||||
test_get_attribute(force_compat_mode, is_open);
|
||||
test_get_attribute(fs_misc_api_check_aggressive, is_open);
|
||||
test_get_attribute(parallel_io, is_open);
|
||||
// test_get_attribute(profile_nvtx is_open);
|
||||
test_get_attribute(allow_system_memory, is_open);
|
||||
test_get_attribute(use_pcip2pdma, is_open);
|
||||
test_get_attribute(prefer_io_uring, is_open);
|
||||
test_get_attribute(force_odirect_mode, is_open);
|
||||
test_get_attribute(skip_topology_detection, is_open);
|
||||
test_get_attribute(stream_memops_bypass, is_open);
|
||||
|
||||
// CUfileDriverStatusFlags
|
||||
test_get_attribute(has_luster_support, is_open);
|
||||
test_get_attribute(has_wekafs_support, is_open);
|
||||
test_get_attribute(has_nfs_support, is_open);
|
||||
test_get_attribute(has_gpfs_support, is_open);
|
||||
test_get_attribute(has_nvme_support, is_open);
|
||||
test_get_attribute(has_nvmeof_support, is_open);
|
||||
test_get_attribute(has_scsi_support, is_open);
|
||||
test_get_attribute(has_scaleflux_csd_support, is_open);
|
||||
test_get_attribute(has_nvmesh_support, is_open);
|
||||
test_get_attribute(has_beegfs_support, is_open);
|
||||
test_get_attribute(has_nvme_p2p_support, is_open);
|
||||
test_get_attribute(has_scatefs_support, is_open);
|
||||
|
||||
// CUfileFeatureFlags
|
||||
test_get_attribute(has_dynamic_routing_support, is_open);
|
||||
test_get_attribute(has_batch_io_support, is_open);
|
||||
test_get_attribute(has_streams_support, is_open);
|
||||
test_get_attribute(has_parallel_io_support, is_open);
|
||||
}
|
||||
|
||||
template <class Attr>
|
||||
void test_set_attribute(const Attr& attr, const bool is_open)
|
||||
{
|
||||
using ValueType = typename Attr::type;
|
||||
|
||||
STATIC_REQUIRE(!noexcept(cudax::cufile_driver.set_attribute(attr, cuda::std::declval<ValueType>())));
|
||||
STATIC_REQUIRE(
|
||||
cuda::std::is_same_v<void, decltype(cudax::cufile_driver.set_attribute(attr, cuda::std::declval<ValueType>()))>);
|
||||
|
||||
const bool expect_success = is_open == Attr::__can_be_set_when_opened || !is_open == Attr::__can_be_set_when_closed;
|
||||
|
||||
ValueType value{};
|
||||
if (expect_success)
|
||||
{
|
||||
CHECK_NOTHROW(value = cudax::cufile_driver.attribute(attr));
|
||||
}
|
||||
|
||||
if (expect_success)
|
||||
{
|
||||
CHECK_NOTHROW(cudax::cufile_driver.set_attribute(attr, value));
|
||||
}
|
||||
else
|
||||
{
|
||||
CHECK_THROWS_AS(cudax::cufile_driver.set_attribute(attr, value), std::runtime_error);
|
||||
}
|
||||
}
|
||||
|
||||
void test_set_attributes(const bool is_open)
|
||||
{
|
||||
using namespace cudax::cufile_driver_attributes;
|
||||
|
||||
// CUFileSizeTConfigParameter_t
|
||||
test_get_attribute(max_io_queue_depth, is_open);
|
||||
test_get_attribute(max_io_threads, is_open);
|
||||
test_get_attribute(min_io_threshold_size_kb, is_open);
|
||||
test_get_attribute(max_request_parallelism, is_open);
|
||||
test_get_attribute(max_direct_io_size_kb, is_open);
|
||||
test_get_attribute(max_device_cache_size_kb, is_open);
|
||||
test_get_attribute(per_buffer_cache_size_kb, is_open);
|
||||
test_get_attribute(max_device_pinned_mem_size_kb, is_open);
|
||||
test_get_attribute(io_batchsize, is_open);
|
||||
test_get_attribute(pollthreshold_size_kb, is_open);
|
||||
test_get_attribute(batch_io_timeout_ms, is_open);
|
||||
|
||||
// CUFileBoolConfigParameter_t
|
||||
test_get_attribute(use_poll_mode, is_open);
|
||||
test_get_attribute(allow_compat_mode, is_open);
|
||||
test_get_attribute(force_compat_mode, is_open);
|
||||
test_get_attribute(fs_misc_api_check_aggressive, is_open);
|
||||
test_get_attribute(parallel_io, is_open);
|
||||
// test_get_attribute(profile_nvtx is_open);
|
||||
test_get_attribute(allow_system_memory, is_open);
|
||||
test_get_attribute(use_pcip2pdma, is_open);
|
||||
test_get_attribute(prefer_io_uring, is_open);
|
||||
test_get_attribute(force_odirect_mode, is_open);
|
||||
test_get_attribute(skip_topology_detection, is_open);
|
||||
test_get_attribute(stream_memops_bypass, is_open);
|
||||
|
||||
// CUfileDriverStatusFlags are unsupported
|
||||
// CUfileFeatureFlags are unsupported
|
||||
}
|
||||
|
||||
void test_open()
|
||||
{
|
||||
STATIC_REQUIRE(!noexcept(cudax::cufile_driver.open()));
|
||||
STATIC_REQUIRE(cuda::std::is_same_v<void, decltype(cudax::cufile_driver.open())>);
|
||||
|
||||
CHECK_NOTHROW(cudax::cufile_driver.open());
|
||||
}
|
||||
|
||||
void test_close()
|
||||
{
|
||||
STATIC_REQUIRE(!noexcept(cudax::cufile_driver.close()));
|
||||
STATIC_REQUIRE(cuda::std::is_same_v<void, decltype(cudax::cufile_driver.close())>);
|
||||
|
||||
CHECK_NOTHROW(cudax::cufile_driver.close());
|
||||
}
|
||||
|
||||
C2H_CCCLRT_TEST("cuFile driver", "[cufile][driver]")
|
||||
{
|
||||
// 1. Test that the driver initial state is closed.
|
||||
test_is_open(false);
|
||||
|
||||
// 2. Test that the attribute ranges can be queried when the driver is closed.
|
||||
test_attribute_ranges(false);
|
||||
|
||||
// 3. Test getting attribute values when the driver is closed.
|
||||
test_get_attributes(false);
|
||||
|
||||
// 4. Test setting attribute values when the driver is closed.
|
||||
test_set_attributes(false);
|
||||
|
||||
// 5. Open the driver.
|
||||
test_open();
|
||||
|
||||
// 6. Test that the driver is opened.
|
||||
test_is_open(true);
|
||||
|
||||
// 7. Test that the attribute ranges can be queried when the driver is opened.
|
||||
test_attribute_ranges(true);
|
||||
|
||||
// 8. Test getting attribute values when the driver is opened.
|
||||
test_get_attributes(true);
|
||||
|
||||
// 9. Test setting attribute values when the driver is opened.
|
||||
test_set_attributes(true);
|
||||
|
||||
// 10. Close the driver.
|
||||
test_close();
|
||||
|
||||
// 11. Test that the driver is closed.
|
||||
test_is_open(false);
|
||||
|
||||
// 12. Test that the attribute ranges can be queried when the driver after is closed.
|
||||
test_attribute_ranges(false);
|
||||
|
||||
// 13. Test getting attribute values when the driver after is closed.
|
||||
test_get_attributes(false);
|
||||
|
||||
// 14. Test setting attribute values when the driver after is closed.
|
||||
test_set_attributes(false);
|
||||
|
||||
// 15. Reopen the driver.
|
||||
test_open();
|
||||
|
||||
// 16. Test that the driver is opened.
|
||||
test_is_open(true);
|
||||
|
||||
// 17. Reopen the driver for second time.
|
||||
test_open();
|
||||
|
||||
// 18. Test that the driver is opened.
|
||||
test_is_open(true);
|
||||
|
||||
// 19. Close the driver.
|
||||
test_close();
|
||||
|
||||
// 20. Test that the driver is closed.
|
||||
test_is_open(false);
|
||||
|
||||
// 21. Close the driver again.
|
||||
test_close();
|
||||
|
||||
// 22. Test that the driver is closed.
|
||||
test_is_open(false);
|
||||
}
|
||||
Reference in New Issue
Block a user