NVIDIA/nvbench
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvbench.git synced 2026-04-28 02:31:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

Execute benchmarks on all devices.

Browse Source
This commit is contained in:
Allison Vacanti
2021-02-12 20:02:20 -05:00
parent 5348f65e12
commit 92cc3b1189
17 changed files with 534 additions and 323 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
testing/create.cu
View File

@@ -83,18 +83,15 @@ std::string run_and_get_state_string(nvbench::benchmark_base &bench,
std::size_t num_type_configs,
std::size_t states_per_type_config)
{
bench.set_devices(std::vector<int>{});
bench.run();
fmt::memory_buffer buffer;
const auto &states = bench.get_states();
ASSERT(states.size() == num_type_configs);
for (const auto &inner_states : states)
ASSERT(states.size() == num_type_configs * states_per_type_config);
for (const auto &state : states)
{
ASSERT(inner_states.size() == states_per_type_config);
for (const auto &state : inner_states)
{
ASSERT(state.is_skipped());
fmt::format_to(buffer, "{}\n", state.get_skip_reason());
}
ASSERT(state.is_skipped());
fmt::format_to(buffer, "{}\n", state.get_skip_reason());
}
return fmt::to_string(buffer);
}