NVIDIA/nvbench
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvbench.git synced 2026-03-14 20:27:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Introduce nvbench::stop_runner_loop exception. If application throws it, runner loop is stopped and other pending benchmark instances are skipped

Browse Source
This commit is contained in:
Oleksandr Pavlyk
2025-12-05 19:32:55 -06:00
parent c286199adc
commit e57f1ecf4c
1 changed files with 20 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
nvbench/runner.cuh
View File

@@ -27,6 +27,13 @@
namespace nvbench
{
struct stop_runner_loop : std::runtime_error
{
// ask compiler to generate all constructor signatures
// that are defined for the base class
using std::runtime_error::runtime_error;
};
// Non-templated code goes here to reduce instantiation costs:
struct runner_base
{
@@ -88,7 +95,8 @@ private:
[&self = *this, &states = m_benchmark.m_states, &type_config_index, &device](
auto type_config_wrapper) {
// Get current type_config:
using type_config = typename decltype(type_config_wrapper)::type;
using type_config = typename decltype(type_config_wrapper)::type;
bool skip_remaining = false;
// Find states with the current device / type_config
for (nvbench::state &cur_state : states)
@@ -99,13 +107,21 @@ private:
self.run_state_prologue(cur_state);
try
{
auto kernel_generator_copy = self.m_kernel_generator;
kernel_generator_copy(cur_state, type_config{});
if (cur_state.is_skipped())
if (!skip_remaining)
{
auto kernel_generator_copy = self.m_kernel_generator;
kernel_generator_copy(cur_state, type_config{});
}
if (skip_remaining || cur_state.is_skipped())
{
self.print_skip_notification(cur_state);
}
}
catch (nvbench::stop_runner_loop &e)
{
skip_remaining = true;
self.handle_sampling_exception(e, cur_state);
}
catch (std::exception &e)
{
self.handle_sampling_exception(e, cur_state);