NVIDIA/nvbench
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvbench.git synced 2026-04-20 06:48:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Allow benchmarks to be specified by index with --benchmark.

Browse Source
This commit is contained in:
Allison Vacanti
2021-02-10 21:40:39 -05:00
parent 2561816f15
commit 3bc8291b28
2 changed files with 39 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
nvbench/option_parser.cu
View File

@@ -357,7 +357,19 @@ void option_parser::print_list() const
void option_parser::add_benchmark(const std::string &name)
{
const auto &mgr = nvbench::benchmark_manager::get();
m_benchmarks.push_back(mgr.get_benchmark(name).clone());
std::unique_ptr<nvbench::benchmark_base> new_bench;
nvbench::int64_t idx{-1};
try
{
::parse(name, idx);
}
catch (std::invalid_argument &)
{}
m_benchmarks.push_back(idx >= 0 ? mgr.get_benchmark(idx).clone()
: mgr.get_benchmark(name).clone());
}
void option_parser::update_axis(const std::string &spec)

34
testing/option_parser.cu
View File

@@ -153,10 +153,19 @@ void test_benchmark_long() // --benchmark
| 8 | 8 | U8 | F64 | 42 | 8 | 3.14 | 'S1' |
)expected";
nvbench::option_parser parser;
parser.parse({"--benchmark", "TestBench"});
const auto test = parser_to_state_string(parser);
ASSERT_MSG(test == ref, "Expected:\n\"{}\"\n\nActual:\n\"{}\"", ref, test);
{
nvbench::option_parser parser;
parser.parse({"--benchmark", "TestBench"});
const auto test = parser_to_state_string(parser);
ASSERT_MSG(test == ref, "Expected:\n\"{}\"\n\nActual:\n\"{}\"", ref, test);
}
{
nvbench::option_parser parser;
parser.parse({"--benchmark", "1"});
const auto test = parser_to_state_string(parser);
ASSERT_MSG(test == ref, "Expected:\n\"{}\"\n\nActual:\n\"{}\"", ref, test);
}
}
void test_benchmark_short() // -b
@@ -175,10 +184,19 @@ void test_benchmark_short() // -b
| 8 | 8 | U8 | F64 | 42 | 8 | 3.14 | 'S1' |
)expected";
nvbench::option_parser parser;
parser.parse({"-b", "TestBench"});
const auto test = parser_to_state_string(parser);
ASSERT_MSG(test == ref, "Expected:\n\"{}\"\n\nActual:\n\"{}\"", ref, test);
{
nvbench::option_parser parser;
parser.parse({"-b", "TestBench"});
const auto test = parser_to_state_string(parser);
ASSERT_MSG(test == ref, "Expected:\n\"{}\"\n\nActual:\n\"{}\"", ref, test);
}
{
nvbench::option_parser parser;
parser.parse({"-b", "1"});
const auto test = parser_to_state_string(parser);
ASSERT_MSG(test == ref, "Expected:\n\"{}\"\n\nActual:\n\"{}\"", ref, test);
}
}
void test_int64_axis_single()