NVIDIA/nvbench
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvbench.git synced 2026-04-26 17:58:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix get_config_count for CPU-only benchmarks. (#218)

Browse Source
This commit is contained in:
Allison Piper
2025-05-01 12:34:35 -04:00
committed by GitHub
parent 433376fd83
commit 9d189280de
2 changed files with 13 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
testing/benchmark.cu
View File

@@ -18,6 +18,7 @@
#include <nvbench/benchmark.cuh>
#include <nvbench/callable.cuh>
#include <nvbench/device_manager.cuh>
#include <nvbench/named_values.cuh>
#include <nvbench/state.cuh>
#include <nvbench/type_list.cuh>
@@ -27,6 +28,7 @@
#include <fmt/format.h>
#include <algorithm>
#include <cstdint>
#include <utility>
#include <variant>
#include <vector>
@@ -279,6 +281,7 @@ void test_clone()
void test_get_config_count()
{
lots_of_types_bench bench;
bench.set_devices(nvbench::device_manager::get().get_devices());
bench.set_type_axes_names({"Integer", "Float", "Other"});
bench.get_axes().get_type_axis(0).set_active_inputs({"I16", "I32"}); // 2, 2
bench.get_axes().get_type_axis(1).set_active_inputs({"F32", "F64"}); // 2, 4
@@ -288,9 +291,13 @@ void test_get_config_count()
bench.add_string_axis("baz", {"str", "ing"}); // 2, 72
bench.add_string_axis("baz", {"single"}); // 1, 72
auto const num_devices = bench.get_devices().size();
auto const num_devices = std::max(std::size_t(1), bench.get_devices().size());
ASSERT_MSG(bench.get_config_count() == 72 * num_devices, "Got {}", bench.get_config_count());
// Check that zero devices (e.g. CPU-only) is the same as a single device:
bench.set_devices(std::vector<int>{});
ASSERT_MSG(bench.get_config_count() == 72, "Got {}", bench.get_config_count());
}
int main()