Skip to content

Commit a2cd7be

Browse files
[BATCH] support ov::enable_profiling (openvinotoolkit#23265)
### Details: To complete task-I added PERF_COUNT to the plugin.cpp And To complete the task-2 I have added get_property_params() # Tickets: openvinotoolkit#23075, CVS-130236
1 parent 5cfefd7 commit a2cd7be

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed

src/plugins/auto_batch/src/plugin.cpp

+3-1
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,9 @@
1919
namespace ov {
2020
namespace autobatch_plugin {
2121

22-
std::vector<std::string> supported_configKeys = {ov::device::priorities.name(), ov::auto_batch_timeout.name()};
22+
std::vector<std::string> supported_configKeys = {ov::device::priorities.name(),
23+
ov::auto_batch_timeout.name(),
24+
ov::enable_profiling.name()};
2325

2426
inline ov::AnyMap merge_properties(ov::AnyMap config, const ov::AnyMap& user_config) {
2527
for (auto&& kvp : user_config) {

src/plugins/auto_batch/tests/unit/plugin_get_property_test.cpp

+2
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,7 @@ class GetPropertyTest : public ::testing::TestWithParam<get_property_params> {
4141
ON_CALL(*m_plugin, get_property).WillByDefault([this](const std::string& name, const ov::AnyMap& arguments) {
4242
return m_plugin->Plugin::get_property(name, arguments);
4343
});
44+
ON_CALL(*m_plugin, get_property(StrEq("PERF_COUNT"), _)).WillByDefault(Return(true));
4445
}
4546
};
4647

@@ -63,6 +64,7 @@ const std::vector<get_property_params> get_property_params_test = {
6364
get_property_params{ov::device::priorities.name(), true},
6465
get_property_params{ov::cache_dir.name(), true},
6566
get_property_params{ov::hint::performance_mode.name(), true},
67+
get_property_params{ov::enable_profiling.name(), false},
6668
};
6769

6870
INSTANTIATE_TEST_SUITE_P(smoke_AutoBatch_BehaviorTests,

0 commit comments

Comments
 (0)