mirror of
https://github.com/NVIDIA/nvbench.git
synced 2026-04-20 14:58:54 +00:00
9f9c6e5278
The old implementation was scattered and ad hoc. This one is slightly less so. More importantly, refactoring to this design will make it easier to add device traversal.
134 lines
4.1 KiB
Plaintext
134 lines
4.1 KiB
Plaintext
#include <nvbench/named_values.cuh>
|
|
|
|
#include "test_asserts.cuh"
|
|
|
|
#include <algorithm>
|
|
|
|
void test_empty()
|
|
{
|
|
nvbench::named_values vals;
|
|
ASSERT(vals.get_size() == 0);
|
|
ASSERT(vals.get_names().size() == 0);
|
|
ASSERT(vals.has_value("Nope") == false);
|
|
ASSERT_THROWS_ANY([[maybe_unused]] auto val = vals.get_value("Nope"));
|
|
ASSERT_THROWS_ANY([[maybe_unused]] auto type = vals.get_type("Nope"));
|
|
// Removing non-existent entries shouldn't cause a problem:
|
|
vals.remove_value("Nope");
|
|
}
|
|
|
|
void test_basic()
|
|
{
|
|
auto sort = [](auto &&vec) {
|
|
std::sort(vec.begin(), vec.end());
|
|
return std::forward<decltype(vec)>(vec);
|
|
};
|
|
|
|
nvbench::named_values vals;
|
|
vals.set_int64("Int", 32);
|
|
vals.set_float64("Float", 34.5);
|
|
vals.set_string("String", "string!");
|
|
vals.set_value("IntVar", {nvbench::int64_t{36}});
|
|
|
|
std::vector<std::string> names{"Float", "Int", "IntVar", "String"};
|
|
|
|
ASSERT(vals.get_size() == 4);
|
|
ASSERT(sort(vals.get_names()) == names);
|
|
|
|
ASSERT(vals.has_value("Float"));
|
|
ASSERT(vals.has_value("Int"));
|
|
ASSERT(vals.has_value("IntVar"));
|
|
ASSERT(vals.has_value("String"));
|
|
|
|
ASSERT(std::get<nvbench::float64_t>(vals.get_value("Float")) == 34.5);
|
|
ASSERT(std::get<nvbench::int64_t>(vals.get_value("Int")) == 32);
|
|
ASSERT(std::get<nvbench::int64_t>(vals.get_value("IntVar")) == 36);
|
|
ASSERT(std::get<std::string>(vals.get_value("String")) == "string!");
|
|
|
|
ASSERT(vals.get_type("Float") == nvbench::named_values::type::float64);
|
|
ASSERT(vals.get_type("Int") == nvbench::named_values::type::int64);
|
|
ASSERT(vals.get_type("IntVar") == nvbench::named_values::type::int64);
|
|
ASSERT(vals.get_type("String") == nvbench::named_values::type::string);
|
|
|
|
ASSERT(vals.get_int64("Int") == 32);
|
|
ASSERT(vals.get_int64("IntVar") == 36);
|
|
ASSERT_THROWS_ANY([[maybe_unused]] auto v = vals.get_int64("Float"));
|
|
ASSERT_THROWS_ANY([[maybe_unused]] auto v = vals.get_int64("String"));
|
|
|
|
ASSERT(vals.get_float64("Float") == 34.5);
|
|
ASSERT_THROWS_ANY([[maybe_unused]] auto v = vals.get_float64("Int"));
|
|
ASSERT_THROWS_ANY([[maybe_unused]] auto v = vals.get_float64("IntVar"));
|
|
ASSERT_THROWS_ANY([[maybe_unused]] auto v = vals.get_float64("String"));
|
|
|
|
ASSERT(vals.get_string("String") == "string!");
|
|
ASSERT_THROWS_ANY([[maybe_unused]] auto v = vals.get_string("Int"));
|
|
ASSERT_THROWS_ANY([[maybe_unused]] auto v = vals.get_string("IntVar"));
|
|
ASSERT_THROWS_ANY([[maybe_unused]] auto v = vals.get_string("Float"));
|
|
|
|
vals.remove_value("IntVar");
|
|
names = {"Float", "Int", "String"};
|
|
|
|
ASSERT(vals.get_size() == 3);
|
|
ASSERT(sort(vals.get_names()) == names);
|
|
|
|
ASSERT(!vals.has_value("IntVar"));
|
|
ASSERT(vals.has_value("Float"));
|
|
ASSERT(vals.has_value("Int"));
|
|
ASSERT(vals.has_value("String"));
|
|
|
|
vals.clear();
|
|
names = {};
|
|
|
|
ASSERT(vals.get_size() == 0);
|
|
ASSERT(sort(vals.get_names()) == names);
|
|
|
|
ASSERT(!vals.has_value("IntVar"));
|
|
ASSERT(!vals.has_value("Float"));
|
|
ASSERT(!vals.has_value("Int"));
|
|
ASSERT(!vals.has_value("String"));
|
|
}
|
|
|
|
void test_append()
|
|
{
|
|
nvbench::named_values vals1;
|
|
vals1.set_int64("Int1", 32);
|
|
vals1.set_float64("Float1", 34.5);
|
|
vals1.set_string("String1", "string1!");
|
|
vals1.set_value("IntVar1", {nvbench::int64_t{36}});
|
|
|
|
nvbench::named_values vals2;
|
|
vals2.set_int64("Int2", 42);
|
|
vals2.set_float64("Float2", 3.14);
|
|
vals2.set_string("String2", "string2!");
|
|
vals2.set_value("IntVar2", {nvbench::int64_t{55}});
|
|
|
|
vals1.append(vals2);
|
|
|
|
// Order should be preserved:
|
|
const auto &names = vals1.get_names();
|
|
ASSERT(names.size() == 8);
|
|
ASSERT(names[0] == "Int1");
|
|
ASSERT(names[1] == "Float1");
|
|
ASSERT(names[2] == "String1");
|
|
ASSERT(names[3] == "IntVar1");
|
|
ASSERT(names[4] == "Int2");
|
|
ASSERT(names[5] == "Float2");
|
|
ASSERT(names[6] == "String2");
|
|
ASSERT(names[7] == "IntVar2");
|
|
|
|
ASSERT(vals1.get_int64("Int1") == 32);
|
|
ASSERT(vals1.get_float64("Float1") == 34.5);
|
|
ASSERT(vals1.get_string("String1") == "string1!");
|
|
ASSERT(vals1.get_int64("IntVar1") == 36);
|
|
ASSERT(vals1.get_int64("Int2") == 42);
|
|
ASSERT(vals1.get_float64("Float2") == 3.14);
|
|
ASSERT(vals1.get_string("String2") == "string2!");
|
|
ASSERT(vals1.get_int64("IntVar2") == 55);
|
|
}
|
|
|
|
int main()
|
|
{
|
|
test_empty();
|
|
test_basic();
|
|
test_append();
|
|
}
|