wip

oneapi-src · Jan 23, 2025 · 1ce6d4e · 1ce6d4e
1 parent e96d2c5
commit 1ce6d4e
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/test/conformance/device/urDeviceGetInfo.cpp b/test/conformance/device/urDeviceGetInfo.cpp
@@ -1968,6 +1968,8 @@ TEST_P(urDeviceGetInfoTest, SuccessESIMDSupported) {
 }
 
 TEST_P(urDeviceGetInfoTest, SuccessComponentDevices) {
+  UUR_KNOWN_FAILURE_ON(uur::LevelZero{}, uur::LevelZeroV2{});
+
   const ur_device_info_t property_name = UR_DEVICE_INFO_COMPONENT_DEVICES;
 
   uint32_t device_count = 0;
@@ -2006,11 +2008,18 @@ TEST_P(urDeviceGetInfoTest, SuccessComponentDevices) {
         ASSERT_NE(componentDevice, nullptr);
       }
     } else {
-      size_t size = 999;
+      size_t property_size = 999;
+
       ASSERT_SUCCESS_OR_OPTIONAL_QUERY(
-          urDeviceGetInfo(device, property_name, 0, nullptr, &size),
+          urDeviceGetInfo(device, property_name, 0, nullptr, &property_size),
           property_name);
-      ASSERT_EQ(size, 0);
+      ASSERT_EQ(property_size, 0);
+
+      std::vector<ur_device_handle_t> componentDevices(property_size);
+      ASSERT_SUCCESS(urDeviceGetInfo(device, property_name, property_size,
+                                     componentDevices.data(), nullptr));
+
+      ASSERT_TRUE(componentDevices.empty());
     }
   }
 }