From ede9b95f533e3709eb006549de629555c33ee9cd Mon Sep 17 00:00:00 2001 From: Patryk Kaminski Date: Thu, 15 Feb 2024 15:21:13 +0100 Subject: [PATCH] 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. --- test/fuzz/urFuzz.cpp | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 deletions(-) diff --git a/test/fuzz/urFuzz.cpp b/test/fuzz/urFuzz.cpp index 0d24f47516..f91ca23633 100644 --- 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; }