Add `batches` options in the graph examples
Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Change-Id: Ib0442a25cb60793f530e1cff5e2aaf167d4a6950
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5812
Reviewed-by: Michalis Spyrou <michalis.spyrou@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
diff --git a/utils/CommonGraphOptions.cpp b/utils/CommonGraphOptions.cpp
index 6ee7ed1..c0270726 100644
--- a/utils/CommonGraphOptions.cpp
+++ b/utils/CommonGraphOptions.cpp
@@ -117,6 +117,7 @@
CommonGraphOptions::CommonGraphOptions(CommandLineParser &parser)
: help(parser.add_option<ToggleOption>("help")),
threads(parser.add_option<SimpleOption<int>>("threads", 1)),
+ batches(parser.add_option<SimpleOption<int>>("batches", 1)),
target(),
data_type(),
data_layout(),
@@ -168,6 +169,7 @@
help->set_help("Show this help message");
threads->set_help("Number of threads to use");
+ batches->set_help("Number of batches to use for the inputs");
target->set_help("Target to execute on");
data_type->set_help("Data type to use");
data_layout->set_help("Data layout to use");
@@ -197,6 +199,7 @@
CommonGraphParams common_params;
common_params.help = options.help->is_set() ? options.help->value() : false;
common_params.threads = options.threads->value();
+ common_params.batches = options.batches->value();
common_params.target = options.target->value();
common_params.data_type = options.data_type->value();
if(options.data_layout->is_set())
diff --git a/utils/CommonGraphOptions.h b/utils/CommonGraphOptions.h
index f2cbd48..afdb78b 100644
--- a/utils/CommonGraphOptions.h
+++ b/utils/CommonGraphOptions.h
@@ -94,6 +94,7 @@
{
bool help{ false };
int threads{ 0 };
+ int batches{ 1 };
arm_compute::graph::Target target{ arm_compute::graph::Target::NEON };
arm_compute::DataType data_type{ DataType::F32 };
arm_compute::DataLayout data_layout{ DataLayout::NHWC };
@@ -151,6 +152,7 @@
ToggleOption *help; /**< Show help option */
SimpleOption<int> *threads; /**< Number of threads option */
+ SimpleOption<int> *batches; /**< Number of batches */
EnumOption<arm_compute::graph::Target> *target; /**< Graph execution target */
EnumOption<arm_compute::DataType> *data_type; /**< Graph data type */
EnumOption<arm_compute::DataLayout> *data_layout; /**< Graph data layout */