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
99 lines
4.1 KiB
Plaintext
99 lines
4.1 KiB
Plaintext
//===----------------------------------------------------------------------===//
|
|
//
|
|
// 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 <testing.cuh>
|
|
|
|
C2H_CCCLRT_TEST("cuFile open mode", "[cufile][open_mode]")
|
|
{
|
|
// 1. Test cufile_open_flags properties
|
|
STATIC_REQUIRE(cuda::std::is_scoped_enum_v<cudax::cufile_open_mode>);
|
|
STATIC_REQUIRE(cuda::std::is_same_v<unsigned, cuda::std::underlying_type_t<cudax::cufile_open_mode>>);
|
|
|
|
// 2. Test cufile_oepn_flags values
|
|
STATIC_REQUIRE(cuda::std::to_underlying(cudax::cufile_open_mode::in) == 0x01u);
|
|
STATIC_REQUIRE(cuda::std::to_underlying(cudax::cufile_open_mode::out) == 0x02u);
|
|
STATIC_REQUIRE(cuda::std::to_underlying(cudax::cufile_open_mode::trunc) == 0x04u);
|
|
STATIC_REQUIRE(cuda::std::to_underlying(cudax::cufile_open_mode::noreplace) == 0x08u);
|
|
STATIC_REQUIRE(cuda::std::to_underlying(cudax::cufile_open_mode::direct) == 0x10u);
|
|
|
|
// 3. Test operator|(cufile_open_flags, cufile_open_flags)
|
|
{
|
|
constexpr auto lhs = cudax::cufile_open_mode::in;
|
|
constexpr auto rhs = cudax::cufile_open_mode::out;
|
|
STATIC_REQUIRE(cuda::std::is_same_v<cudax::cufile_open_mode, decltype(lhs | rhs)>);
|
|
STATIC_REQUIRE(noexcept(lhs | rhs));
|
|
STATIC_REQUIRE(cuda::std::to_underlying(lhs | rhs) == 0x3u);
|
|
STATIC_REQUIRE(cuda::std::to_underlying(lhs | lhs) == 0x1u);
|
|
}
|
|
|
|
// 3. Test operator=|(cufile_open_flags&, cufile_open_flags)
|
|
{
|
|
auto lhs = cudax::cufile_open_mode::in;
|
|
auto rhs = cudax::cufile_open_mode::out;
|
|
STATIC_REQUIRE(cuda::std::is_same_v<cudax::cufile_open_mode&, decltype(lhs |= rhs)>);
|
|
STATIC_REQUIRE(noexcept(lhs |= rhs));
|
|
REQUIRE(cuda::std::to_underlying(lhs |= cudax::cufile_open_mode::in) == 0x1u);
|
|
REQUIRE(cuda::std::to_underlying(lhs |= rhs) == 0x3u);
|
|
}
|
|
|
|
// 4. Test operator&(cufile_open_flags, cufile_open_flags)
|
|
{
|
|
constexpr auto lhs = cudax::cufile_open_mode::in;
|
|
constexpr auto rhs = cudax::cufile_open_mode::out;
|
|
STATIC_REQUIRE(cuda::std::is_same_v<cudax::cufile_open_mode, decltype(lhs & rhs)>);
|
|
STATIC_REQUIRE(noexcept(lhs & rhs));
|
|
STATIC_REQUIRE(cuda::std::to_underlying(lhs & lhs) == 0x1u);
|
|
STATIC_REQUIRE(cuda::std::to_underlying(lhs & rhs) == 0x0u);
|
|
}
|
|
|
|
// 5. Test operator=&(cufile_open_flags&, cufile_open_flags)
|
|
{
|
|
auto lhs = cudax::cufile_open_mode::in;
|
|
auto rhs = cudax::cufile_open_mode::out;
|
|
STATIC_REQUIRE(cuda::std::is_same_v<cudax::cufile_open_mode&, decltype(lhs &= rhs)>);
|
|
STATIC_REQUIRE(noexcept(lhs &= rhs));
|
|
REQUIRE(cuda::std::to_underlying(lhs &= cudax::cufile_open_mode::in) == 0x1u);
|
|
REQUIRE(cuda::std::to_underlying(lhs &= rhs) == 0x0u);
|
|
}
|
|
|
|
// 5. Test operator^(cufile_open_flags, cufile_open_flags)
|
|
{
|
|
constexpr auto lhs = cudax::cufile_open_mode::in;
|
|
constexpr auto rhs = cudax::cufile_open_mode::out;
|
|
STATIC_REQUIRE(cuda::std::is_same_v<cudax::cufile_open_mode, decltype(lhs ^ rhs)>);
|
|
STATIC_REQUIRE(noexcept(lhs ^ rhs));
|
|
STATIC_REQUIRE(cuda::std::to_underlying(lhs ^ lhs) == 0x0u);
|
|
STATIC_REQUIRE(cuda::std::to_underlying(lhs ^ rhs) == 0x3u);
|
|
}
|
|
|
|
// 6. Test operator=^(cufile_open_flags&, cufile_open_flags)
|
|
{
|
|
auto lhs = cudax::cufile_open_mode::in;
|
|
auto rhs = cudax::cufile_open_mode::out;
|
|
STATIC_REQUIRE(cuda::std::is_same_v<cudax::cufile_open_mode&, decltype(lhs ^= rhs)>);
|
|
STATIC_REQUIRE(noexcept(lhs ^= rhs));
|
|
REQUIRE(cuda::std::to_underlying(lhs ^= rhs) == 0x3u);
|
|
REQUIRE(cuda::std::to_underlying(lhs ^= cudax::cufile_open_mode::in) == 0x2u);
|
|
}
|
|
|
|
// 5. Test operator~(cufile_open_flags)
|
|
{
|
|
constexpr auto v = cudax::cufile_open_mode::in;
|
|
STATIC_REQUIRE(cuda::std::is_same_v<cudax::cufile_open_mode, decltype(~v)>);
|
|
STATIC_REQUIRE(noexcept(~v));
|
|
STATIC_REQUIRE(cuda::std::to_underlying(~v) == ~0x1u);
|
|
}
|
|
}
|