add performance hint to time infer (#8480)

openvinotoolkit · Nov 10, 2021 · 9fc64ea · 9fc64ea
1 parent 35ba404
commit 9fc64ea
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/tests/time_tests/src/timetests/timetest_infer.cpp b/tests/time_tests/src/timetests/timetest_infer.cpp
@@ -3,6 +3,7 @@
 //
 
 #include <inference_engine.hpp>
+#include <ie_plugin_config.hpp>
 #include <iostream>
 
 #include "common_utils.h"
@@ -29,6 +30,8 @@ int runPipeline(const std::string &model, const std::string &device) {
       {
         SCOPED_TIMER(load_plugin);
         ie.GetVersions(device);
+        // enables performance hint for specified device
+        ie.SetConfig({{CONFIG_KEY(PERFORMANCE_HINT), CONFIG_VALUE(LATENCY)}}, device);
       }
       {
         SCOPED_TIMER(create_exenetwork);

diff --git a/tests/time_tests/src/timetests/timetest_infer_cache.cpp b/tests/time_tests/src/timetests/timetest_infer_cache.cpp
@@ -3,6 +3,7 @@
 //
 
 #include <inference_engine.hpp>
+#include <ie_plugin_config.hpp>
 #include <iostream>
 
 #include "common_utils.h"
@@ -31,7 +32,8 @@ int runPipeline(const std::string &model, const std::string &device) {
       }
       {
         SCOPED_TIMER(load_network);
-        ie.SetConfig({{"CACHE_DIR", "models_cache"}});
+        // enables cache
+        ie.SetConfig({{CONFIG_KEY(CACHE_DIR), "models_cache"}});
         exeNetwork = ie.LoadNetwork(model, device);
       }
       {