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

accomodate the config per device-id, after rebase to the latest master

Browse Source
This commit is contained in:
myshevts 2021-10-21 13:43:26 +03:00
parent 0de3055825
commit e11c6cb0ad
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
inference-engine/src/cldnn_engine/cldnn_engine.cpp
View File

@ -680,7 +680,8 @@ Parameter clDNNEngine::GetMetric(const std::string& name, const std::map<std::st
IE_SET_METRIC_RETURN(GPU_UARCH_VERSION, s.str());
} else if (name == METRIC_KEY(OPTIMAL_BATCH)) {
auto network = options.find("MODEL_ADDRESS")->second.as<InferenceEngine::CNNNetwork const*>();
auto networkCloned = CloneAndTransformNetwork(*network, _impl->m_config);
Config config = _impl->m_configs.GetConfig(device_id);
auto networkCloned = CloneAndTransformNetwork(*network, config);
// DG1
const float L3_cache_size = 12*1024*1024;
unsigned int batch = 1;