Store default engine index per root device

Resolves: NEO-5768
Signed-off-by: Mateusz Jablonski <mateusz.jablonski@intel.com>
This commit is contained in:
Mateusz Jablonski
2021-04-20 12:24:04 +00:00
committed by Compute-Runtime-Automation
parent 8da0838ba4
commit 9a22d06efe
7 changed files with 27 additions and 20 deletions

View File

@ -42,6 +42,7 @@ MemoryManager::MemoryManager(ExecutionEnvironment &executionEnvironment) : execu
bool anyLocalMemorySupported = false;
defaultEngineIndex.resize(executionEnvironment.rootDeviceEnvironments.size());
for (uint32_t rootDeviceIndex = 0; rootDeviceIndex < executionEnvironment.rootDeviceEnvironments.size(); ++rootDeviceIndex) {
auto hwInfo = executionEnvironment.rootDeviceEnvironments[rootDeviceIndex]->getHardwareInfo();
localMemoryUsageBankSelector.emplace_back(new LocalMemoryUsageBankSelector(HwHelper::getSubDevicesCount(hwInfo)));