Fix fuzz test failing to create a device

Changes from #019551c in urDeviceGet validation resulted in a failure to create devices successfully in fuzz tests.
oneapi-src · Feb 15, 2024 · 918b0aa · 918b0aa
1 parent f11823e
commit 918b0aa
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/test/fuzz/urFuzz.cpp b/test/fuzz/urFuzz.cpp
@@ -38,15 +38,19 @@ int ur_device_get(TestState &state) {
         return -1;
     }
 
-    ur_result_t res = urDeviceGet(state.platforms[state.platform_num],
-                                  state.device_type, state.num_entries,
-                                  state.devices.data(), &state.num_devices);
+    ur_result_t res = UR_RESULT_SUCCESS;
+    if (state.devices.size() == 0) {
+        res = urDeviceGet(state.platforms[state.platform_num],
+                          state.device_type, 0, nullptr, &state.num_devices);
+        state.devices.resize(state.num_devices);
+    } else {
+        res =
+            urDeviceGet(state.platforms[state.platform_num], state.device_type,
+                        state.num_entries, state.devices.data(), nullptr);
+    }
     if (res != UR_RESULT_SUCCESS) {
         return -1;
     }
-    if (state.devices.size() != state.num_devices) {
-        state.devices.resize(state.num_devices);
-    }
 
     return 0;
 }