diff options
Diffstat (limited to 'tests/benchmark_examples')
-rw-r--r-- | tests/benchmark_examples/RunExample.cpp | 15 |
1 files changed, 12 insertions, 3 deletions
diff --git a/tests/benchmark_examples/RunExample.cpp b/tests/benchmark_examples/RunExample.cpp index 05430e7628..81c879ff2d 100644 --- a/tests/benchmark_examples/RunExample.cpp +++ b/tests/benchmark_examples/RunExample.cpp @@ -59,17 +59,26 @@ public: void do_setup() override { ARM_COMPUTE_ERROR_ON_NULLPTR(g_example.get()); - g_example->do_setup(g_example_argv.size(), &g_example_argv[0]); + _is_setup = g_example->do_setup(g_example_argv.size(), &g_example_argv[0]); } void do_run() override { - g_example->do_run(); + if(_is_setup) + { + g_example->do_run(); + } } void do_teardown() override { - g_example->do_teardown(); + if(_is_setup) + { + g_example->do_teardown(); + } g_example = nullptr; } + +private: + bool _is_setup{ false }; }; int run_example(int argc, char **argv, std::unique_ptr<Example> example) |