[UR] Return spills for all ZE kernels in UR kernel via v1 level_zero plugin.

Author: kurapov-peter

unified-runtime/source/adapters/level_zero/kernel.cpp

@@ -752,10 +752,24 @@ ur_result_t urKernelGetInfo(
   case UR_KERNEL_INFO_NUM_ARGS:
     return ReturnValue(uint32_t{Kernel->ZeKernelProperties->numKernelArgs});
   case UR_KERNEL_INFO_SPILL_MEM_SIZE: {
-    std::vector<uint32_t> spills = {
-        uint32_t{Kernel->ZeKernelProperties->spillMemSize}};
-    return ReturnValue(static_cast<const uint32_t *>(spills.data()),
-                       spills.size());
+    try {
+      std::vector<uint32_t> Spills;
+      Spills.reserve(Kernel->ZeKernels.size());
+      for (auto &ZeKernel : Kernel->ZeKernels) {
+        ze_kernel_properties_t props;
+        props.stype = ZE_STRUCTURE_TYPE_KERNEL_PROPERTIES;
+        props.pNext = nullptr;
+        ZE2UR_CALL(zeKernelGetProperties, (ZeKernel, &props));
+        uint32_t spillMemSize = props.spillMemSize;
+        Spills.push_back(spillMemSize);
+      }
+      return ReturnValue(static_cast<const uint32_t *>(Spills.data()),
+                       Spills.size());
+    } catch (const std::bad_alloc &) {
+      return UR_RESULT_ERROR_OUT_OF_HOST_MEMORY;
+    } catch (...) {
+      return UR_RESULT_ERROR_UNKNOWN;
+    }
   }
   case UR_KERNEL_INFO_REFERENCE_COUNT:
     return ReturnValue(uint32_t{Kernel->RefCount.load()});