Change the API to return an array of uint32_t to fix the L0 v2 implementation

Author: kurapov-peter

include/ur_api.h

@@ -5997,7 +5997,7 @@ typedef enum ur_kernel_info_t {
   /// [uint32_t][optional-query] Return the number of registers used by the
   /// compiled kernel.
   UR_KERNEL_INFO_NUM_REGS = 6,
-  /// [uint32_t][optional-query] Return the spill memory size allocated by
+  /// [uint32_t[]][optional-query] Return the spill memory size allocated by
   /// the compiler.
   UR_KERNEL_INFO_SPILL_MEM_SIZE = 7,
   /// @cond

include/ur_print.hpp

@@ -8876,17 +8876,18 @@ inline ur_result_t printTagged(std::ostream &os, const void *ptr,
     os << ")";
   } break;
   case UR_KERNEL_INFO_SPILL_MEM_SIZE: {
-    const uint32_t *tptr = (const uint32_t *)ptr;
-    if (sizeof(uint32_t) > size) {
-      os << "invalid size (is: " << size << ", expected: >=" << sizeof(uint32_t)
-         << ")";
-      return UR_RESULT_ERROR_INVALID_SIZE;
-    }
-    os << (const void *)(tptr) << " (";
 
-    os << *tptr;
+    const uint32_t *tptr = (const uint32_t *)ptr;
+    os << "{";
+    size_t nelems = size / sizeof(uint32_t);
+    for (size_t i = 0; i < nelems; ++i) {
+      if (i != 0) {
+        os << ", ";
+      }
 
-    os << ")";
+      os << tptr[i];
+    }
+    os << "}";
   } break;
   default:
     os << "unknown enumerator";

scripts/core/kernel.yml

@@ -126,7 +126,7 @@ etors:
     - name: NUM_REGS
       desc: "[uint32_t][optional-query] Return the number of registers used by the compiled kernel."
     - name: SPILL_MEM_SIZE
-      desc: "[uint32_t][optional-query] Return the spill memory size allocated by the compiler."
+      desc: "[uint32_t[]][optional-query] Return the spill memory size allocated by the compiler."
 --- #--------------------------------------------------------------------------
 type: enum
 desc: "Get Kernel Work Group information"

source/adapters/level_zero/kernel.cpp

@@ -752,7 +752,9 @@ ur_result_t urKernelGetInfo(
   case UR_KERNEL_INFO_NUM_ARGS:
     return ReturnValue(uint32_t{Kernel->ZeKernelProperties->numKernelArgs});
   case UR_KERNEL_INFO_SPILL_MEM_SIZE:
-    return ReturnValue(uint32_t{Kernel->ZeKernelProperties->spillMemSize});
+    return ReturnValue(static_cast<const uint32_t *>(
+                           &(Kernel->ZeKernelProperties->spillMemSize)),
+                       1ul);
   case UR_KERNEL_INFO_REFERENCE_COUNT:
     return ReturnValue(uint32_t{Kernel->RefCount.load()});
   case UR_KERNEL_INFO_ATTRIBUTES:

source/adapters/level_zero/v2/kernel.cpp

@@ -624,8 +624,16 @@ ur_result_t urKernelGetInfo(ur_kernel_handle_t hKernel,
   case UR_KERNEL_INFO_NUM_REGS:
   case UR_KERNEL_INFO_NUM_ARGS:
     return ReturnValue(uint32_t{hKernel->getCommonProperties().numKernelArgs});
-  case UR_KERNEL_INFO_SPILL_MEM_SIZE:
-    return ReturnValue(uint32_t{hKernel->getProperties().spillMemSize});
+  case UR_KERNEL_INFO_SPILL_MEM_SIZE: {
+    std::shared_lock<ur_shared_mutex> Guard(hKernel->getProgramHandle()->Mutex);
+    auto devices = hKernel->getProgramHandle()->AssociatedDevices;
+    std::vector<uint32_t> spills(devices.size());
+    for (size_t i = 0; i < spills.size(); ++i) {
+      spills[i] = uint32_t{hKernel->getProperties(devices[i]).spillMemSize};
+    }
+    return ReturnValue(static_cast<const uint32_t *>(spills.data()),
+                       spills.size());
+  }
   case UR_KERNEL_INFO_REFERENCE_COUNT:
     return ReturnValue(uint32_t{hKernel->RefCount.load()});
   case UR_KERNEL_INFO_ATTRIBUTES: {