[SYCL] Fix descendent device support for composite/component devices (#13513)

Fixed a case where it was not possible to create a queue for a component
device using context for a composite device.

Author: AlexeySachkov

sycl/source/detail/context_impl.hpp

@@ -186,8 +186,17 @@ class context_impl {
       return hasDevice(Device);
 
     while (!hasDevice(Device)) {
-      if (Device->isRootDevice())
+      if (Device->isRootDevice()) {
+        if (Device->has(aspect::ext_oneapi_is_component)) {
+          // Component devices should be implicitly usable in context created
+          // for a composite device they belong to.
+          auto CompositeDevice = Device->get_info<
+              ext::oneapi::experimental::info::device::composite_device>();
+          return hasDevice(detail::getSyclObjImpl(CompositeDevice));
+        }
+
         return false;
+      }
       Device = detail::getSyclObjImpl(
           Device->get_info<info::device::parent_device>());
     }

sycl/unittests/Extensions/CompositeDevice.cpp

@@ -79,7 +79,7 @@ pi_result after_piContextCreate(const pi_context_properties *,
 
 } // namespace
 
-TEST(CompositeDeviceTest, DescendentDeviceSupport) {
+TEST(CompositeDeviceTest, DescendentDeviceSupportInContext) {
   sycl::unittest::PiMock Mock(sycl::backend::ext_oneapi_level_zero);
   Mock.redefine<sycl::detail::PiApiKind::piDevicesGet>(redefine_piDevicesGet);
   Mock.redefineAfter<sycl::detail::PiApiKind::piDeviceGetInfo>(
@@ -119,3 +119,25 @@ TEST(CompositeDeviceTest, DescendentDeviceSupport) {
   ASSERT_EQ(CompositeDevContext.get_devices().size(), 1u);
   ASSERT_EQ(CompositeDevContext.get_devices().front(), CompositeDevice);
 }
+
+TEST(CompositeDeviceTest, DescendentDeviceSupportInQueue) {
+  sycl::unittest::PiMock Mock(sycl::backend::ext_oneapi_level_zero);
+  Mock.redefine<sycl::detail::PiApiKind::piDevicesGet>(redefine_piDevicesGet);
+  Mock.redefineAfter<sycl::detail::PiApiKind::piDeviceGetInfo>(
+      after_piDeviceGetInfo);
+  Mock.redefineAfter<sycl::detail::PiApiKind::piContextCreate>(
+      after_piContextCreate);
+
+  sycl::platform Plt = Mock.getPlatform();
+  ASSERT_EQ(Plt.get_backend(), sycl::backend::ext_oneapi_level_zero);
+
+  sycl::device ComponentDevice = Plt.get_devices()[0];
+  ASSERT_TRUE(ComponentDevice.has(sycl::aspect::ext_oneapi_is_component));
+
+  auto CompositeDevice = ComponentDevice.get_info<
+      sycl::ext::oneapi::experimental::info::device::composite_device>();
+  sycl::context CompositeDevContext(CompositeDevice);
+  // Component device should be implicitly usable as part of composite context,
+  // so there should be no errors during queue creation below.
+  sycl::queue Queue(CompositeDevContext, ComponentDevice);
+}