Implement DeviceKernelInfo for interop kernels

Author: vinser52

sycl/include/sycl/handler.hpp

@@ -501,14 +501,7 @@ class __SYCL_EXPORT handler {
     setType(detail::CGType::Kernel);
   }
 
-  void setDeviceKernelInfo(kernel &&Kernel) {
-    MKernel = detail::getSyclObjImpl(std::move(Kernel));
-    MKernelName = getKernelName();
-    setType(detail::CGType::Kernel);
-
-    // If any extra actions are added here make sure that logic around
-    // `lambdaAndKernelHaveEqualName` calls can handle that.
-  }
+  void setDeviceKernelInfo(kernel &&Kernel);
 
 #ifndef __INTEL_PREVIEW_BREAKING_CHANGES
   // TODO: Those functions are not used anymore, remove it in the next

sycl/source/detail/kernel_impl.cpp

@@ -26,7 +26,17 @@ kernel_impl::kernel_impl(Managed<ur_kernel_handle_t> &&Kernel,
       MCreatedFromSource(true),
       MKernelBundleImpl(KernelBundleImpl ? KernelBundleImpl->shared_from_this()
                                          : nullptr),
-      MIsInterop(true), MKernelArgMaskPtr{ArgMask} {
+      MIsInterop(true), MKernelArgMaskPtr{ArgMask},
+      MDeviceKernelInfo(
+          MIsInterop
+              ? DeviceKernelInfoUniquePtr(
+                    new DeviceKernelInfo(
+                        CompileTimeKernelInfoTy{std::string_view(getName())}),
+                    DeviceKernelInfoDeleter)
+              : DeviceKernelInfoUniquePtr(
+                    &ProgramManager::getInstance().getOrCreateDeviceKernelInfo(
+                        KernelNameStrT(getName())),
+                    noopDeviceKernelInfoDeleter)) {
   ur_context_handle_t UrContext = nullptr;
   // Using the adapter from the passed ContextImpl
   getAdapter().call<UrApiKind::urKernelGetInfo>(
@@ -52,7 +62,17 @@ kernel_impl::kernel_impl(Managed<ur_kernel_handle_t> &&Kernel,
       MDeviceImageImpl(std::move(DeviceImageImpl)),
       MKernelBundleImpl(KernelBundleImpl.shared_from_this()),
       MIsInterop(MDeviceImageImpl->getOriginMask() & ImageOriginInterop),
-      MKernelArgMaskPtr{ArgMask}, MCacheMutex{CacheMutex} {
+      MKernelArgMaskPtr{ArgMask}, MCacheMutex{CacheMutex},
+      MDeviceKernelInfo(
+          MIsInterop
+              ? DeviceKernelInfoUniquePtr(
+                    new DeviceKernelInfo(
+                        CompileTimeKernelInfoTy{std::string_view(getName())}),
+                    DeviceKernelInfoDeleter)
+              : DeviceKernelInfoUniquePtr(
+                    &ProgramManager::getInstance().getOrCreateDeviceKernelInfo(
+                        KernelNameStrT(getName())),
+                    noopDeviceKernelInfoDeleter)) {
   // Enable USM indirect access for interop and non-sycl-jit source kernels.
   // sycl-jit kernels will enable this if needed through the regular kernel
   // path.

sycl/source/detail/kernel_impl.hpp

@@ -238,6 +238,10 @@ class kernel_impl {
   std::mutex *getCacheMutex() const { return MCacheMutex; }
   std::string_view getName() const;
 
+  DeviceKernelInfo *getDeviceKernelInfoPtr() const {
+    return MDeviceKernelInfo.get();
+  }
+
 private:
   Managed<ur_kernel_handle_t> MKernel;
   const std::shared_ptr<context_impl> MContext;
@@ -251,6 +255,17 @@ class kernel_impl {
   std::mutex *MCacheMutex = nullptr;
   mutable std::string MName;
 
+  using DeviceKernelInfoDeleterTy = void (*)(DeviceKernelInfo *);
+  using DeviceKernelInfoUniquePtr =
+      std::unique_ptr<DeviceKernelInfo, DeviceKernelInfoDeleterTy>;
+  DeviceKernelInfoUniquePtr MDeviceKernelInfo;
+
+  static void DeviceKernelInfoDeleter(DeviceKernelInfo *Ptr) { delete Ptr; }
+
+  static void noopDeviceKernelInfoDeleter(DeviceKernelInfo *) {
+    // do nothing
+  }
+
   bool isBuiltInKernel(device_impl &Device) const;
   void checkIfValidForNumArgsInfoQuery() const;
 

sycl/source/handler.cpp

@@ -2539,6 +2539,16 @@ bool handler::HasAssociatedAccessor(detail::AccessorImplHost *Req,
 void handler::setType(sycl::detail::CGType Type) { impl->MCGType = Type; }
 sycl::detail::CGType handler::getType() const { return impl->MCGType; }
 
+void handler::setDeviceKernelInfo(kernel &&Kernel) {
+  MKernel = detail::getSyclObjImpl(std::move(Kernel));
+  MKernelName = getKernelName();
+  setDeviceKernelInfoPtr(MKernel->getDeviceKernelInfoPtr());
+  setType(detail::CGType::Kernel);
+
+  // If any extra actions are added here make sure that logic around
+  // `lambdaAndKernelHaveEqualName` calls can handle that.
+}
+
 #ifndef __INTEL_PREVIEW_BREAKING_CHANGES
 void handler::setNDRangeDescriptorPadded(sycl::range<3> N,
                                          bool SetNumWorkGroups, int Dims) {