intel/compute-runtime
1
0
Fork 0
mirror of https://github.com/intel/compute-runtime.git synced 2026-01-05 09:09:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: bindless image from usm pooled ptr

Browse Source 
Add handling for creating bindless image from usm pooled ptr.
Make relevant ULTs test USM device pool enabled/disabled.

Related-To: NEO-6893

Signed-off-by: Dominik Dabek <dominik.dabek@intel.com>
This commit is contained in:
Dominik Dabek
2025-05-28 14:25:06 +00:00
committed by Compute-Runtime-Automation
parent ce1bcb536c
commit cdb088b477
2 changed files with 59 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
level_zero/core/source/image/image_hw.inl
View File

@@ -18,6 +18,7 @@
#include "shared/source/memory_manager/allocation_properties.h"
#include "shared/source/memory_manager/memory_manager.h"
#include "shared/source/memory_manager/unified_memory_manager.h"
#include "shared/source/memory_manager/unified_memory_pooling.h"
#include "shared/source/release_helper/release_helper.h"
#include "level_zero/core/source/device/device.h"
@@ -105,6 +106,8 @@ ze_result_t ImageCoreFamily<gfxCoreFamily>::initialize(Device *device, const ze_
this->samplerDesc.pNext = nullptr;
}
NEO::UsmMemAllocPool *usmPool = nullptr;
if (!isImageView()) {
if (lookupTable.isSharedHandle) {
if (!lookupTable.sharedHandleType.isSupportedHandle) {
@@ -139,6 +142,15 @@ ze_result_t ImageCoreFamily<gfxCoreFamily>::initialize(Device *device, const ze_
if (usmAllocation == nullptr) {
return ZE_RESULT_ERROR_INVALID_ARGUMENT;
}
if (this->device->getNEODevice()->getUsmMemAllocPool() &&
this->device->getNEODevice()->getUsmMemAllocPool()->isInPool(lookupTable.imageProperties.pitchedPtr)) {
usmPool = this->device->getNEODevice()->getUsmMemAllocPool();
if (nullptr == usmPool->getPooledAllocationBasePtr(lookupTable.imageProperties.pitchedPtr)) {
return ZE_RESULT_ERROR_INVALID_ARGUMENT;
}
}
allocation = usmAllocation->gpuAllocations.getGraphicsAllocation(device->getRootDeviceIndex());
}
}
@@ -157,10 +169,14 @@ ze_result_t ImageCoreFamily<gfxCoreFamily>::initialize(Device *device, const ze_
imgInfo.rowPitch = imgInfo.imgDesc.imageWidth * imgInfo.surfaceFormat->imageElementSizeInBytes;
}
imgInfo.slicePitch = imgInfo.rowPitch * imgInfo.imgDesc.imageHeight;
imgInfo.size = allocation->getUnderlyingBufferSize();
imgInfo.qPitch = 0;
UNRECOVERABLE_IF(imgInfo.offset != 0);
if (!isImageView()) {
imgInfo.size = allocation->getUnderlyingBufferSize();
if (usmPool) {
imgInfo.size = usmPool->getPooledAllocationSize(lookupTable.imageProperties.pitchedPtr);
imgInfo.offset = usmPool->getOffsetInPool(lookupTable.imageProperties.pitchedPtr);
}
}
}
if (this->bindlessImage) {