Add UR_L0_V2_DISABLE_ZE_LAUNCH_KERNEL_WITH_ARGS debugging variable (#20703)

Add `UR_L0_V2_DISABLE_ZE_LAUNCH_KERNEL_WITH_ARGS` debugging environment
variable to disable calling
`ZeCommandListAppendLaunchKernelWithArguments()`.

---------

Signed-off-by: Lukasz Dorau <[email protected]>

Author: ldorau

source/adapters/level_zero/platform.cpp

@@ -538,6 +538,10 @@ ur_result_t ur_platform_handle_t_::initialize() {
       .DriverSupportsCooperativeKernelLaunchWithArgs =
       this->isDriverVersionNewerOrSimilar(1, 6, 35005);
 
+  ZeCommandListAppendLaunchKernelWithArgumentsExt
+      .DisableZeLaunchKernelWithArgs =
+      getenv_tobool("UR_L0_V2_DISABLE_ZE_LAUNCH_KERNEL_WITH_ARGS", false);
+
   return UR_RESULT_SUCCESS;
 }
 

source/adapters/level_zero/platform.hpp

@@ -166,5 +166,6 @@ struct ur_platform_handle_t_ : ur::handle_base<ur::level_zero::ddi_getter>,
   struct ZeCommandListAppendLaunchKernelWithArgumentsExtension {
     bool Supported = false;
     bool DriverSupportsCooperativeKernelLaunchWithArgs = false;
+    bool DisableZeLaunchKernelWithArgs = false;
   } ZeCommandListAppendLaunchKernelWithArgumentsExt;
 };

source/adapters/level_zero/v2/command_list_manager.cpp

@@ -1256,8 +1256,11 @@ ur_result_t ur_command_list_manager::appendKernelLaunchWithArgsExp(
   bool CooperativeCompatible =
       hPlatform->ZeCommandListAppendLaunchKernelWithArgumentsExt
           .DriverSupportsCooperativeKernelLaunchWithArgs;
+  bool DisableZeLaunchKernelWithArgs =
+      hPlatform->ZeCommandListAppendLaunchKernelWithArgumentsExt
+          .DisableZeLaunchKernelWithArgs;
   bool RunNewPath =
-      KernelWithArgsSupported &&
+      !DisableZeLaunchKernelWithArgs && KernelWithArgsSupported &&
       (!cooperativeKernelLaunchRequested ||
        (cooperativeKernelLaunchRequested && CooperativeCompatible));
   if (RunNewPath) {