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

return device infer request in passthrough mode (#11253)

Browse Source 
Signed-off-by: fishbell <bell.song@intel.com>
This commit is contained in:
yanlan song 2022-03-30 09:21:13 +08:00 committed by GitHub
parent 4145291e84
commit f6e5ec9684
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/plugins/auto/executable_network.cpp
View File

@ -91,6 +91,8 @@ MultiDeviceExecutableNetwork::MultiDeviceExecutableNetwork(const DeviceMap<Infer
auto& network = networkValue.second;
GenerateWorkers(device, network);
}
if (_networksPerDevice.size() == 1)
_passthroughExeNet = _networksPerDevice.begin()->second;
}
void MultiDeviceExecutableNetwork::GenerateWorkers(const std::string& device, const SoExecutableNetworkInternal& executableNetwork) {
@ -328,6 +330,7 @@ MultiDeviceExecutableNetwork::MultiDeviceExecutableNetwork(const std::string&
} else {
// only one device need to load network, do not need to load it async
_loadContext[ACTUALDEVICE].task();
_passthroughExeNet = _loadContext[ACTUALDEVICE].executableNetwork;
}
WaitFirstNetworkReady();
}
@ -706,6 +709,11 @@ InferenceEngine::IInferRequestInternal::Ptr MultiDeviceExecutableNetwork::Create
}
IInferRequestInternal::Ptr MultiDeviceExecutableNetwork::CreateInferRequest() {
if (_passthroughExeNet) {
auto res = _passthroughExeNet->CreateInferRequest();
res->setPointerToExecutableNetworkInternal(shared_from_this());
return res;
}
IInferRequestInternal::Ptr syncRequestImpl;
if (this->_plugin) {
const auto& core = _plugin->GetCore();

1
src/plugins/auto/executable_network.hpp
View File

@ -165,6 +165,7 @@ private:
unsigned int _cpuHelpInferCount = 0;
double _cpuHelpFps = 0.0;
Time _cpuHelpReleaseTime;
InferenceEngine::SoExecutableNetworkInternal _passthroughExeNet;
};
} // namespace MultiDevicePlugin