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
84 lines
2.1 KiB
C++
84 lines
2.1 KiB
C++
// SPDX-FileCopyrightText: Copyright (c) 2011-2024, NVIDIA CORPORATION. All rights reserved.
|
|
// SPDX-License-Identifier: BSD-3
|
|
|
|
#pragma once
|
|
|
|
#include <cuda/std/tuple>
|
|
|
|
#include <chrono>
|
|
#include <iostream>
|
|
#include <string>
|
|
|
|
// #define C2H_DEBUG_TIMING
|
|
|
|
#ifdef C2H_DEBUG_TIMING
|
|
# define C2H_TIME_SECTION_INIT() [[maybe_unused]] c2h::cpu_timer _c2h_timer_
|
|
# define C2H_TIME_SECTION_RESET() _c2h_timer_.reset()
|
|
# define C2H_TIME_SECTION(label) _c2h_timer_.print_elapsed_seconds_and_reset(label)
|
|
# define C2H_TIME_SCOPE(label) [[maybe_unused]] c2h::scoped_cpu_timer _c2h_scoped_cpu_timer_(label)
|
|
#else
|
|
# define C2H_TIME_SECTION_INIT() /* no-op */ []() {}()
|
|
# define C2H_TIME_SECTION_RESET() /* no-op */ []() {}()
|
|
# define C2H_TIME_SECTION(label) /* no-op */ []() {}()
|
|
# define C2H_TIME_SCOPE(label) /* no-op */ []() {}()
|
|
#endif
|
|
|
|
namespace c2h
|
|
{
|
|
class cpu_timer
|
|
{
|
|
std::chrono::high_resolution_clock::time_point m_start;
|
|
|
|
public:
|
|
cpu_timer()
|
|
: m_start(std::chrono::high_resolution_clock::now())
|
|
{}
|
|
|
|
void reset()
|
|
{
|
|
m_start = std::chrono::high_resolution_clock::now();
|
|
}
|
|
|
|
int elapsed_ms() const
|
|
{
|
|
auto duration = std::chrono::high_resolution_clock::now() - m_start;
|
|
auto ms = std::chrono::duration_cast<std::chrono::milliseconds>(duration);
|
|
return static_cast<int>(ms.count());
|
|
}
|
|
|
|
std::uint64_t elapsed_us() const
|
|
{
|
|
auto duration = std::chrono::high_resolution_clock::now() - m_start;
|
|
auto us = std::chrono::duration_cast<std::chrono::microseconds>(duration);
|
|
return static_cast<std::uint64_t>(us.count());
|
|
}
|
|
|
|
void print_elapsed_seconds(const std::string& label)
|
|
{
|
|
printf("%0.6f s: %s\n", static_cast<float>(this->elapsed_us()) / 1000000.f, label.c_str());
|
|
}
|
|
|
|
void print_elapsed_seconds_and_reset(const std::string& label)
|
|
{
|
|
this->print_elapsed_seconds(label);
|
|
this->reset();
|
|
}
|
|
};
|
|
|
|
class scoped_cpu_timer
|
|
{
|
|
cpu_timer m_timer;
|
|
std::string m_label;
|
|
|
|
public:
|
|
explicit scoped_cpu_timer(std::string label)
|
|
: m_label(std::move(label))
|
|
{}
|
|
|
|
~scoped_cpu_timer()
|
|
{
|
|
m_timer.print_elapsed_seconds(m_label);
|
|
}
|
|
};
|
|
} // namespace c2h
|