Correct default engine index in MemoryManager

Related-To: NEO-5610
Signed-off-by: Maciej Dziuban <maciej.dziuban@intel.com>
This commit is contained in:
Maciej Dziuban
2021-04-19 15:06:08 +00:00
committed by Compute-Runtime-Automation
parent 5318ff1872
commit 927c097b0e
3 changed files with 41 additions and 1 deletions

View File

@@ -83,7 +83,15 @@ bool Device::createDeviceImpl() {
}
}
executionEnvironment->memoryManager->setDefaultEngineIndex(defaultEngineIndex);
uint32_t defaultEngineIndexWithinMemoryManager = 0;
for (auto engineIndex = 0u; engineIndex < executionEnvironment->memoryManager->getRegisteredEnginesCount(); engineIndex++) {
OsContext *engine = executionEnvironment->memoryManager->getRegisteredEngines()[engineIndex].osContext;
if (engine == getDefaultEngine().osContext) {
defaultEngineIndexWithinMemoryManager = engineIndex;
break;
}
}
executionEnvironment->memoryManager->setDefaultEngineIndex(defaultEngineIndexWithinMemoryManager);
auto osInterface = getRootDeviceEnvironment().osInterface.get();