Fix v2 L0 adapter

Author: againull

source/adapters/level_zero/program.hpp

@@ -10,8 +10,12 @@
 #pragma once
 
 #include "common.hpp"
-#include "context.hpp"
 #include "device.hpp"
+#ifdef UR_ADAPTER_LEVEL_ZERO_V2
+#include "v2/context.hpp"
+#else
+#include "context.hpp"
+#endif
 
 struct ur_program_handle_t_ : _ur_object {
   // ur_program_handle_t_() {}

source/adapters/level_zero/ur_level_zero.hpp

@@ -25,9 +25,17 @@
 #include <zes_api.h>
 
 #include "common.hpp"
+#ifdef UR_ADAPTER_LEVEL_ZERO_V2
+#include "v2/context.hpp"
+#else
 #include "context.hpp"
+#endif
 #include "device.hpp"
+#ifdef UR_ADAPTER_LEVEL_ZERO_V2
+#include "v2/event.hpp"
+#else
 #include "event.hpp"
+#endif
 #include "image.hpp"
 #include "kernel.hpp"
 #include "memory.hpp"

source/adapters/level_zero/v2/kernel.cpp

@@ -40,7 +40,14 @@ ur_kernel_handle_t_::ur_kernel_handle_t_(ur_program_handle_t hProgram,
       deviceKernels(hProgram->Context->getPlatform()->getNumDevices()) {
   ur::level_zero::urProgramRetain(hProgram);
 
-  for (auto [zeDevice, zeModule] : hProgram->ZeModuleMap) {
+  for (auto &Dev : hProgram->AssociatedDevices) {
+    auto zeDevice = Dev->ZeDevice;
+    // Program may be associated with all devices from the context but built
+    // only for subset of devices.
+    if (hProgram->getState(zeDevice) != ur_program_handle_t_::state::Exe)
+      continue;
+
+    auto zeModule = hProgram->getZeModuleHandle(zeDevice);
     ZeStruct<ze_kernel_desc_t> zeKernelDesc;
     zeKernelDesc.pKernelName = kernelName;