mirror of
https://github.com/NVIDIA/nvbench.git
synced 2026-04-19 22:38:52 +00:00
e302583c67
Invoking a variadic macro with zero variadic args is illegal until C++20. The extra calls to fmt::format were unnecessary, anyway.
41 lines
967 B
Plaintext
41 lines
967 B
Plaintext
#include <nvbench/cuda_timer.cuh>
|
|
|
|
#include <nvbench/cuda_stream.cuh>
|
|
#include <nvbench/types.cuh>
|
|
|
|
#include "sleep_kernel.cuh"
|
|
#include "test_asserts.cuh"
|
|
|
|
#include <fmt/format.h>
|
|
|
|
void test_basic(cudaStream_t time_stream,
|
|
cudaStream_t exec_stream,
|
|
bool expected)
|
|
{
|
|
nvbench::cuda_timer timer;
|
|
|
|
NVBENCH_CUDA_CALL(cudaDeviceSynchronize());
|
|
|
|
timer.start(time_stream);
|
|
sleep_kernel<<<1, 1, 0, exec_stream>>>(0.25);
|
|
timer.stop(time_stream);
|
|
|
|
NVBENCH_CUDA_CALL(cudaDeviceSynchronize());
|
|
const bool captured = timer.get_duration() > 0.25;
|
|
ASSERT_MSG(captured == expected,
|
|
"Unexpected result from timer: {} seconds (expected {})",
|
|
timer.get_duration(),
|
|
(expected ? "> 0.25s" : "< 0.25s"));
|
|
}
|
|
|
|
void test_basic()
|
|
{
|
|
nvbench::cuda_stream stream1;
|
|
nvbench::cuda_stream stream2;
|
|
|
|
test_basic(stream1, stream1, true);
|
|
test_basic(stream1, stream2, false);
|
|
}
|
|
|
|
int main() { test_basic(); }
|