From 9fbc8211d48f9cfa34f01fc46d44f898d1710872 Mon Sep 17 00:00:00 2001
From: "Ptak, Slawomir" <slawomir.ptak@intel.com>
Date: Mon, 1 Dec 2025 11:34:19 +0000
Subject: [PATCH] [SYCL] Update the KernelCopy unit test to use the kernel
 launch with args

The KernelArgsTest.KernelCopy unit test used the urKernelSetArgValue
function to verify the arguments passed to the kernel. Since the use
of that function was replaced with urEnqueueKernelLaunchWithArgsExp,
that unit test needed an update.
---
 .../unittests/kernel-and-program/KernelArgs.cpp | 17 ++++++++++-------
 1 file changed, 10 insertions(+), 7 deletions(-)

diff --git a/sycl/unittests/kernel-and-program/KernelArgs.cpp b/sycl/unittests/kernel-and-program/KernelArgs.cpp
index b4cd8bf8af89c..618551b10d578 100644
--- a/sycl/unittests/kernel-and-program/KernelArgs.cpp
+++ b/sycl/unittests/kernel-and-program/KernelArgs.cpp
@@ -37,7 +37,7 @@ struct KernelInfo<TestKernelWithIntPtr> : public unittest::MockKernelInfoBase {
 
 private:
   static constexpr detail::kernel_param_desc_t IntParamDesc = {
-      detail::kernel_param_kind_t::kind_std_layout, 0, 0};
+      detail::kernel_param_kind_t::kind_std_layout, sizeof(int), 0};
 };
 
 } // namespace detail
@@ -50,11 +50,13 @@ static sycl::unittest::MockDeviceImageArray<1> ImgArray{&Img};
 
 static int ArgInt = 123;
 
-ur_result_t redefined_urKernelSetArgValue(void *pParams) {
-  auto params = *static_cast<ur_kernel_set_arg_value_params_t *>(pParams);
+ur_result_t redefined_EnqueueKernelLaunchWithArgsExp(void *pParams) {
+  auto params =
+      *static_cast<ur_enqueue_kernel_launch_with_args_exp_params_t *>(pParams);
+  auto args = *params.ppArgs;
+  int value = *static_cast<const int *>(args[0].value.value);
 
-  int ArgValue = *static_cast<const int *>(*params.ppArgValue);
-  EXPECT_EQ(ArgValue, ArgInt);
+  EXPECT_EQ(value, ArgInt);
 
   return UR_RESULT_SUCCESS;
 }
@@ -77,8 +79,9 @@ void runKernelWithArgs(queue &Queue, int ArgI) {
 // of the kernel lambda is deallocated.
 TEST(KernelArgsTest, KernelCopy) {
   sycl::unittest::UrMock<> Mock;
-  mock::getCallbacks().set_before_callback("urKernelSetArgValue",
-                                           &redefined_urKernelSetArgValue);
+  mock::getCallbacks().set_before_callback(
+      "urEnqueueKernelLaunchWithArgsExp",
+      &redefined_EnqueueKernelLaunchWithArgsExp);
 
   platform Plt = sycl::platform();