mirror of
https://github.com/intel/compute-runtime.git
synced 2025-12-27 07:44:16 +08:00
Refactor clSharedMemAllocINTEL
- Choose allocation path according to given device, - Connect device in createUnifiedAllocationWithDeviceStorage - Fix type in clGetMemAllocInfoINTEL Change-Id: I9e743001b4c032a712c939c2917f16de0a61b100 Signed-off-by: Krzysztof Gibala <krzysztof.gibala@intel.com>
This commit is contained in:
committed by
sys_ocldev
parent
6112e99329
commit
c9a2406ea6
@@ -3525,7 +3525,7 @@ void *clSharedMemAllocINTEL(
|
||||
|
||||
ErrorCodeHelper err(errcodeRet, CL_SUCCESS);
|
||||
|
||||
auto retVal = validateObjects(WithCastToInternal(context, &neoContext), WithCastToInternal(device, &neoDevice));
|
||||
auto retVal = validateObjects(WithCastToInternal(context, &neoContext));
|
||||
|
||||
if (retVal != CL_SUCCESS) {
|
||||
err.set(retVal);
|
||||
@@ -3547,8 +3547,13 @@ void *clSharedMemAllocINTEL(
|
||||
}
|
||||
|
||||
unifiedMemoryProperties.device = device;
|
||||
unifiedMemoryProperties.subdeviceBitfield = neoDevice->getDefaultEngine().osContext->getDeviceBitfield();
|
||||
if (!device) {
|
||||
return neoContext->getSVMAllocsManager()->createUnifiedMemoryAllocation(neoContext->getDevice(0)->getRootDeviceIndex(), size, unifiedMemoryProperties);
|
||||
}
|
||||
|
||||
validateObjects(WithCastToInternal(device, &neoDevice));
|
||||
|
||||
unifiedMemoryProperties.subdeviceBitfield = neoDevice->getDefaultEngine().osContext->getDeviceBitfield();
|
||||
return neoContext->getSVMAllocsManager()->createSharedUnifiedMemoryAllocation(neoContext->getDevice(0)->getRootDeviceIndex(), size, unifiedMemoryProperties, neoContext->getSpecialQueue());
|
||||
}
|
||||
|
||||
@@ -3626,7 +3631,7 @@ cl_int clGetMemAllocInfoINTEL(
|
||||
return retVal;
|
||||
}
|
||||
case CL_MEM_ALLOC_FLAGS_INTEL: {
|
||||
retVal = info.set<uint32_t>(unifiedMemoryAllocation->allocationFlagsProperty.allAllocFlags);
|
||||
retVal = info.set<cl_mem_alloc_flags_intel>(unifiedMemoryAllocation->allocationFlagsProperty.allAllocFlags);
|
||||
return retVal;
|
||||
}
|
||||
case CL_MEM_ALLOC_DEVICE_INTEL: {
|
||||
|
||||
Reference in New Issue
Block a user