[SYCL] Add caching when using interop constructor (#3327)

When interop constructor is used it creates new impl every time. This patch allows to use already created impls in the constructor.

Signed-off-by: mdimakov <[email protected]>

Author: maximdimakov

sycl/source/device.cpp

@@ -30,13 +30,17 @@ void force_type(info::device_type &t, const info::device_type &ft) {
 
 device::device() : impl(detail::device_impl::getHostDeviceImpl()) {}
 
-device::device(cl_device_id deviceId)
-    : impl(std::make_shared<detail::device_impl>(
-          detail::pi::cast<pi_native_handle>(deviceId),
-          RT::getPlugin<backend::opencl>())) {
+device::device(cl_device_id DeviceId) {
   // The implementation constructor takes ownership of the native handle so we
   // must retain it in order to adhere to SYCL 1.2.1 spec (Rev6, section 4.3.1.)
-  clRetainDevice(deviceId);
+  detail::RT::PiDevice Device;
+  auto Plugin = detail::RT::getPlugin<backend::opencl>();
+  Plugin.call<detail::PiApiKind::piextDeviceCreateWithNativeHandle>(
+      detail::pi::cast<pi_native_handle>(DeviceId), nullptr, &Device);
+  auto Platform =
+      detail::platform_impl::getPlatformFromPiDevice(Device, Plugin);
+  impl = Platform->getOrMakeDeviceImpl(Device, Platform);
+  clRetainDevice(DeviceId);
 }
 
 device::device(const device_selector &deviceSelector) {

sycl/source/platform.cpp

@@ -19,10 +19,11 @@ namespace sycl {
 
 platform::platform() : impl(detail::platform_impl::getHostPlatformImpl()) {}
 
-platform::platform(cl_platform_id PlatformId)
-    : impl(std::make_shared<detail::platform_impl>(
-          detail::pi::cast<detail::RT::PiPlatform>(PlatformId),
-          RT::getPlugin<backend::opencl>())) {}
+platform::platform(cl_platform_id PlatformId) {
+  impl = detail::platform_impl::getOrMakePlatformImpl(
+      detail::pi::cast<detail::RT::PiPlatform>(PlatformId),
+      detail::RT::getPlugin<backend::opencl>());
+}
 
 platform::platform(const device_selector &dev_selector) {
   *this = dev_selector.select_device().get_platform();