Media
/
compute-runtime
mirror of https://github.com/intel/compute-runtime.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

feature: Add TBX support for USM shared memory with USM host ptr 

Related-To: NEO-11158

Signed-off-by: Milczarek, Slawomir <slawomir.milczarek@intel.com>
This commit is contained in:
Milczarek, Slawomir 2024-04-22 10:42:12 +00:00 committed by Compute-Runtime-Automation
parent ae139aeffd
commit ab2eeeca4c
1 changed files with 24 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
shared/source/memory_manager/os_agnostic_memory_manager.cpp
View File

@ -136,7 +136,30 @@ GraphicsAllocation *OsAgnosticMemoryManager::allocateGraphicsMemoryWithAlignment
} }
GraphicsAllocation *OsAgnosticMemoryManager::allocateUSMHostGraphicsMemory(const AllocationData &allocationData) { GraphicsAllocation *OsAgnosticMemoryManager::allocateUSMHostGraphicsMemory(const AllocationData &allocationData) {
return allocateGraphicsMemoryWithHostPtr(allocationData); AllocationData allocData = allocationData;
if (allocData.type == AllocationType::svmCpu) {
auto &rootDeviceEnvironment = *executionEnvironment.rootDeviceEnvironments[allocData.rootDeviceIndex];
auto &productHelper = rootDeviceEnvironment.getHelper<ProductHelper>();
allocData.alignment = alignUpNonZero<size_t>(allocationData.alignment, productHelper.getSvmCpuAlignment());
allocData.size = alignUp(allocationData.size, allocData.alignment);
}
auto memoryAllocation = allocateGraphicsMemoryWithHostPtr(allocData);
if (memoryAllocation && allocData.type == AllocationType::svmCpu) {
void *gpuPtr = reserveCpuAddressRange(allocData.size, allocData.rootDeviceIndex);
if (nullptr == gpuPtr) {
cleanGraphicsMemoryCreatedFromHostPtr(memoryAllocation);
delete memoryAllocation;
return nullptr;
}
memoryAllocation->setReservedAddressRange(gpuPtr, allocData.size);
gpuPtr = alignUp(gpuPtr, allocData.alignment);
auto cpuPtr = const_cast<void *>(allocData.hostPtr);
auto gmmHelper = getGmmHelper(allocData.rootDeviceIndex);
auto canonizedGpuAddress = gmmHelper->canonize(reinterpret_cast<uint64_t>(gpuPtr));
memoryAllocation->setCpuPtrAndGpuAddress(cpuPtr, canonizedGpuAddress);
}
return memoryAllocation;
} }
GraphicsAllocation *OsAgnosticMemoryManager::allocateGraphicsMemoryForNonSvmHostPtr(const AllocationData &allocationData) { GraphicsAllocation *OsAgnosticMemoryManager::allocateGraphicsMemoryForNonSvmHostPtr(const AllocationData &allocationData) {