[SYCL] Level-Zero support of piextKernelSetArgSampler

Signed-off-by: Sergey V Maslov <[email protected]>

Author: smaslov-intel

sycl/plugins/level_zero/pi_level0.cpp

@@ -2028,8 +2028,12 @@ pi_result piextKernelSetArgMemObj(pi_kernel Kernel, pi_uint32 ArgIndex,
 // Special version of piKernelSetArg to accept pi_sampler.
 pi_result piextKernelSetArgSampler(pi_kernel Kernel, pi_uint32 ArgIndex,
                                    const pi_sampler *ArgValue) {
-  die("piextKernelSetArgSampler: not implemented");
-  return {};
+  assert(Kernel);
+  ZE_CALL(zeKernelSetArgumentValue(
+      pi_cast<ze_kernel_handle_t>(Kernel->ZeKernel),
+      pi_cast<uint32_t>(ArgIndex), sizeof(void *), &(*ArgValue)->ZeSampler));
+
+  return PI_SUCCESS;
 }
 
 pi_result piKernelGetInfo(pi_kernel Kernel, pi_kernel_info ParamName,

sycl/plugins/level_zero/pi_level0.hpp

@@ -346,9 +346,6 @@ struct _pi_sampler : _pi_object {
   _pi_sampler(ze_sampler_handle_t Sampler) : ZeSampler{Sampler} {}
 
   // L0 sampler handle.
-  // TODO: It is important that L0 handler is the first data member. Workaround
-  // in SYCL RT (in ExecCGCommand::enqueueImp()) relies on this. This comment
-  // should be removed when workaround in SYCL runtime will be removed.
   ze_sampler_handle_t ZeSampler;
 };