IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

unify name (#15067)

Browse Source 
Signed-off-by: fishbell <bell.song@intel.com>

Signed-off-by: fishbell <bell.song@intel.com>
This commit is contained in:
yanlan song
2023-01-15 11:03:38 +08:00
committed by GitHub
parent fe290b7acb
commit d79c04a0b1
2 changed files with 15 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/plugins/auto/auto_executable_network.cpp
View File

@@ -165,6 +165,8 @@ IE::Parameter AutoExecutableNetwork::GetMetric(const std::string& name) const {
continue;
} else {
std::string exeDevices_string = _autoSchedule->_loadContext[i].workName.substr(_autoSchedule->_loadContext[i].workName.find(":") + 1);
if (exeDevices_string == "CPU_HELP")
exeDevices_string = "(CPU)";
std::stringstream ss(exeDevices_string);
std::string item;
while (getline(ss, item, ',')) {

15
src/tests/unit/auto/auto_release_helper_test.cpp
View File

@@ -170,15 +170,26 @@ TEST_P(AutoReleaseHelperTest, releaseResource) {
config.insert({InferenceEngine::MultiDeviceConfigParams::KEY_MULTI_DEVICE_PRIORITIES,
CommonTestUtils::DEVICE_CPU + std::string(",") + CommonTestUtils::DEVICE_GPU});
std::shared_ptr<InferenceEngine::IExecutableNetworkInternal> exeNetwork;
if (cpuSuccess || accSuccess)
if (cpuSuccess || accSuccess) {
ASSERT_NO_THROW(exeNetwork = plugin->LoadExeNetworkImpl(cnnNet, config));
else
if (!cpuSuccess)
EXPECT_EQ(exeNetwork->GetMetric(ov::execution_devices.name()).as<std::string>(), CommonTestUtils::DEVICE_GPU);
else
EXPECT_EQ(exeNetwork->GetMetric(ov::execution_devices.name()).as<std::string>(), "(CPU)");
} else {
ASSERT_THROW(exeNetwork = plugin->LoadExeNetworkImpl(cnnNet, config), InferenceEngine::Exception);
}
auto sharedcount = mockExeNetwork._ptr.use_count();
auto requestsharedcount = inferReqInternal.use_count();
std::this_thread::sleep_for(std::chrono::milliseconds(500));
EXPECT_EQ(mockExeNetwork._ptr.use_count(), sharedcount - decreaseCount);
EXPECT_EQ(inferReqInternal.use_count(), requestsharedcount - decreaseCount);
if (cpuSuccess || accSuccess) {
if (accSuccess)
EXPECT_EQ(exeNetwork->GetMetric(ov::execution_devices.name()).as<std::string>(), CommonTestUtils::DEVICE_GPU);
else
EXPECT_EQ(exeNetwork->GetMetric(ov::execution_devices.name()).as<std::string>(), CommonTestUtils::DEVICE_CPU);
}
}
//