WIP 2025-09-12 pią 12:12

Author: lslusarczyk

src/provider/provider_level_zero.c

@@ -361,12 +361,17 @@ umf_result_t umfLevelZeroMemoryProviderParamsSetResidentDevices(
         return UMF_RESULT_ERROR_INVALID_ARGUMENT;
     }
 
-    if (residentDevicesCount && !residentDevicesIndices) {
+    if (residentDevicesCount > 0 && residentDevicesIndices == NULL) {
         LOG_ERR("Resident devices indices array is NULL, but "
                 "residentDevicesCount is not zero");
         return UMF_RESULT_ERROR_INVALID_ARGUMENT;
     }
 
+    if (deviceCount > 0 && hDevices == NULL) {
+        LOG_ERR("All devices array is NULL, but deviceCount is not zero");
+        return UMF_RESULT_ERROR_INVALID_ARGUMENT;
+    }
+
     hParams->device_handles = hDevices;
     hParams->device_count = deviceCount;
     hParams->resident_device_indices = residentDevicesIndices;