Extend zeCommandListAppendMemoryPrefetch to migrate to associated device

Related-To: NEO-6740

Signed-off-by: Milczarek, Slawomir <slawomir.milczarek@intel.com>
This commit is contained in:
Milczarek, Slawomir
2022-03-22 15:54:48 +00:00
committed by Compute-Runtime-Automation
parent 1f7fd45251
commit f03f530327
10 changed files with 50 additions and 13 deletions

View File

@@ -41,7 +41,8 @@ ze_result_t CommandListCoreFamily<IGFX_XE_HPC_CORE>::appendMemoryPrefetch(const
if (memoryManager->isKmdMigrationAvailable(device->getRootDeviceIndex()) &&
(allocData->memoryType == InternalMemoryType::SHARED_UNIFIED_MEMORY)) {
auto alloc = allocData->gpuAllocations.getGraphicsAllocation(device->getRootDeviceIndex());
memoryManager->setMemPrefetch(alloc, device->getRootDeviceIndex());
auto subDeviceId = static_cast<DeviceImp *>(device)->getPhysicalSubDeviceId();
memoryManager->setMemPrefetch(alloc, subDeviceId, device->getRootDeviceIndex());
}
}