@@ -39,6 +39,8 @@ static constexpr char log_level_message[] = "Optional. Log level for OpenVINO li
39
39
40
40
static constexpr char config_message[] = " Optional. Path to the configuration file." ;
41
41
42
+ static constexpr char perf_count_message[] = " Optional. Set PERF_COUNT config property." ;
43
+
42
44
static constexpr char inputs_precision_message[] = " Optional. Specifies precision for all input layers of the network." ;
43
45
44
46
static constexpr char outputs_precision_message[] =
@@ -79,14 +81,15 @@ static const char shape_message[] =
79
81
" For dynamic dimensions use symbol `?` or '-1'. Ex. [?,3,?,?]."
80
82
" For bounded dimensions specify range 'min..max'. Ex. [1..10,3,?,?]." ;
81
83
82
- static const char override_model_batch_size [] = " Enforce a model to be compiled for batch size" ;
84
+ static constexpr char override_model_batch_size_message [] = " Enforce a model to be compiled for batch size" ;
83
85
84
86
DEFINE_bool (h, false , help_message);
85
87
DEFINE_string (m, " " , model_message);
86
88
DEFINE_string (d, " " , targetDeviceMessage);
87
89
DEFINE_string (o, " " , output_message);
88
90
DEFINE_string (log_level, " " , log_level_message);
89
91
DEFINE_string (c, " " , config_message);
92
+ DEFINE_bool (pc, false , perf_count_message);
90
93
DEFINE_string (ip, " " , inputs_precision_message);
91
94
DEFINE_string (op, " " , outputs_precision_message);
92
95
DEFINE_string (iop, " " , iop_message);
@@ -97,7 +100,7 @@ DEFINE_string(iml, "", inputs_model_layout_message);
97
100
DEFINE_string (oml, " " , outputs_model_layout_message);
98
101
DEFINE_string (ioml, " " , ioml_message);
99
102
DEFINE_string (shape, " " , shape_message);
100
- DEFINE_uint32 (override_model_batch_size, 1 , override_model_batch_size );
103
+ DEFINE_uint32 (override_model_batch_size, 1 , override_model_batch_size_message );
101
104
102
105
namespace {
103
106
std::vector<std::string> splitStringList (const std::string& str, char delim) {
@@ -449,6 +452,9 @@ int main(int argc, char* argv[]) {
449
452
auto timeBeforeLoadNetwork = std::chrono::steady_clock::now ();
450
453
std::cout << " Parsing configuration file" << std::endl;
451
454
auto configs = parseConfigFile ();
455
+ if (FLAGS_pc) {
456
+ configs[" PERF_COUNT" ] = " YES" ;
457
+ }
452
458
453
459
std::cout << " Compiling model" << std::endl;
454
460
auto compiledModel = core.compile_model (model, FLAGS_d, {configs.begin (), configs.end ()});
0 commit comments