intel/compute-runtime
1
0
Fork 0
mirror of https://github.com/intel/compute-runtime.git synced 2025-12-21 09:14:47 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: respect arg size when setting kernel arg in OCL path

Browse Source 
Fixes: #777

https://github.com/intel/compute-runtime/issues/777
Signed-off-by: Mateusz Jablonski <mateusz.jablonski@intel.com>
This commit is contained in:
Mateusz Jablonski
2025-02-24 12:23:56 +00:00
committed by Compute-Runtime-Automation
parent 8154c21697
commit c10ff0f3c6
5 changed files with 99 additions and 54 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
opencl/source/kernel/kernel.cpp
View File

@@ -1714,17 +1714,16 @@ cl_int Kernel::setArgImmediate(uint32_t argIndex,
const auto &argAsVal = kernelInfo.kernelDescriptor.payloadMappings.explicitArgs[argIndex].as<ArgDescValue>();
for (const auto &element : argAsVal.elements) {
DEBUG_BREAK_IF(element.size <= 0);
if (static_cast<size_t>(element.sourceOffset + element.size) > argSize) {
return CL_INVALID_ARG_SIZE;
}
auto pDst = ptrOffset(crossThreadData, element.offset);
auto pSrc = ptrOffset(argVal, element.sourceOffset);
DEBUG_BREAK_IF(!(ptrOffset(pDst, element.size) <= crossThreadDataEnd));
auto dstAvailableSpace = crossThreadDataEnd - pDst;
if (element.sourceOffset < argSize) {
size_t maxBytesToCopy = argSize - element.sourceOffset;
size_t bytesToCopy = std::min(static_cast<size_t>(element.size), maxBytesToCopy);
memcpy_s(pDst, element.size, pSrc, bytesToCopy);
}
memcpy_s(pDst, dstAvailableSpace, pSrc, element.size);
}
retVal = CL_SUCCESS;