diff --git a/unit_tests/kernel/kernel_tests.cpp b/unit_tests/kernel/kernel_tests.cpp index d891416b69..e974b77194 100644 --- a/unit_tests/kernel/kernel_tests.cpp +++ b/unit_tests/kernel/kernel_tests.cpp @@ -2693,6 +2693,19 @@ TEST(KernelTest, givenDebugVariableSetWhenKernelHasStatefulBufferAccessThenMarkK } } +TEST(KernelTest, givenKernelWithPairArgumentWhenItIsInitializedThenPatchImmediateIsUsedAsArgHandler) { + HardwareInfo localHwInfo = *platformDevices[0]; + + std::unique_ptr device(MockDevice::createWithNewExecutionEnvironment(&localHwInfo)); + auto context = clUniquePtr(new MockContext(device.get())); + MockKernelWithInternals kernel(*device, context.get()); + kernel.kernelInfo.kernelArgInfo.resize(1); + kernel.kernelInfo.kernelArgInfo.at(0).typeStr = "pair"; + + kernel.mockKernel->initialize(); + EXPECT_EQ(&Kernel::setArgImmediate, kernel.mockKernel->kernelArgHandlers[0]); +} + TEST(KernelTest, whenNullAllocationThenAssignNullPointerToCacheFlushVector) { auto device = std::unique_ptr(MockDevice::createWithNewExecutionEnvironment(platformDevices[0])); MockKernelWithInternals kernel(*device);