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

fix: use correct pool allocation size in tryGetExistingSvmAllocation

Browse Source 
Related-To: NEO-16082

Signed-off-by: Aleksander Czerwionka <aleksander.czerwionka@intel.com>
This commit is contained in:
Aleksander Czerwionka
2025-10-29 10:09:14 +00:00
committed by Compute-Runtime-Automation
parent 08fc0c1e1f
commit d8858e7165
4 changed files with 64 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
opencl/source/context/context.cpp
View File

@@ -122,7 +122,19 @@ cl_int Context::tryGetExistingSvmAllocation(const void *ptr,
SvmAllocationData *svmEntry = getSVMAllocsManager()->getSVMAlloc(ptr);
if (svmEntry) {
memoryType = svmEntry->memoryType;
if ((svmEntry->gpuAllocations.getGraphicsAllocation(rootDeviceIndex)->getGpuAddress() + svmEntry->size) < (castToUint64(ptr) + size)) {
UsmMemAllocPool *pool = nullptr;
if (this->getDevice(0u)->getPlatform()->getHostMemAllocPool().isInPool(ptr)) {
pool = &this->getDevice(0u)->getPlatform()->getHostMemAllocPool();
} else if (this->getDeviceMemAllocPool().isInPool(ptr)) {
pool = &this->getDeviceMemAllocPool();
}
if (pool) {
size_t pooledSize = pool->getPooledAllocationSize(ptr);
uint64_t pooledBasePtr = castToUint64(pool->getPooledAllocationBasePtr(ptr));
if ((pooledBasePtr + pooledSize) < (castToUint64(ptr) + size)) {
return CL_INVALID_OPERATION;
}
} else if ((svmEntry->gpuAllocations.getGraphicsAllocation(rootDeviceIndex)->getGpuAddress() + svmEntry->size) < (castToUint64(ptr) + size)) {
return CL_INVALID_OPERATION;
}
allocation = svmEntry->cpuAllocation ? svmEntry->cpuAllocation : svmEntry->gpuAllocations.getGraphicsAllocation(rootDeviceIndex);