Add method for unknown kernel arg

Signed-off-by: Maciej Plewka <maciej.plewka@intel.com>
This commit is contained in:
Maciej Plewka
2021-04-15 15:17:08 +00:00
committed by Compute-Runtime-Automation
parent 77b39ea3df
commit a51ee31500
4 changed files with 37 additions and 1 deletions

View File

@@ -515,6 +515,10 @@ ze_result_t KernelImp::setArgBufferWithAlloc(uint32_t argIndex, uintptr_t argVal
return ZE_RESULT_SUCCESS;
}
ze_result_t KernelImp::setArgUnknown(uint32_t argIndex, size_t argSize, const void *argVal) {
return ZE_RESULT_SUCCESS;
}
ze_result_t KernelImp::setArgBuffer(uint32_t argIndex, size_t argSize, const void *argVal) {
const auto &allArgs = kernelImmData->getDescriptor().payloadMappings.explicitArgs;
const auto &currArg = allArgs[argIndex];
@@ -671,7 +675,7 @@ ze_result_t KernelImp::initialize(const ze_kernel_desc_t *desc) {
for (const auto &argT : kernelImmData->getDescriptor().payloadMappings.explicitArgs) {
switch (argT.type) {
default:
UNRECOVERABLE_IF(true);
this->kernelArgHandlers.push_back(&KernelImp::setArgUnknown);
break;
case NEO::ArgDescriptor::ArgTPointer:
this->kernelArgHandlers.push_back(&KernelImp::setArgBuffer);

View File

@@ -58,6 +58,8 @@ struct KernelImp : Kernel {
ze_result_t setArgBuffer(uint32_t argIndex, size_t argSize, const void *argVal);
ze_result_t setArgUnknown(uint32_t argIndex, size_t argSize, const void *argVal);
ze_result_t setArgRedescribedImage(uint32_t argIndex, ze_image_handle_t argVal) override;
ze_result_t setArgBufferWithAlloc(uint32_t argIndex, uintptr_t argVal, NEO::GraphicsAllocation *allocation) override;