fix image/buffer max size

OpenCL-CTS is checking that image/buffer max size is below CL_DEVICE_MAX_MEM_ALLOC_SIZE.
kpet · Jan 24, 2025 · 85746d5 · 85746d5
1 parent a91d6a8
commit 85746d5
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/api.cpp b/src/api.cpp
@@ -724,12 +724,15 @@ cl_int CLVK_API_CALL clGetDeviceInfo(cl_device_id dev,
         size_ret = sizeof(val_uint);
         break;
     case CL_DEVICE_IMAGE_MAX_BUFFER_SIZE:
-        val_sizet = device->image_max_buffer_size();
+        val_sizet = std::min(device->image_max_buffer_size(),
+                             device->max_mem_alloc_size());
         copy_ptr = &val_sizet;
         size_ret = sizeof(val_sizet);
         break;
     case CL_DEVICE_IMAGE_MAX_ARRAY_SIZE:
-        val_sizet = device->vulkan_limits().maxImageArrayLayers;
+        val_sizet =
+            std::min((uint64_t)device->vulkan_limits().maxImageArrayLayers,
+                     device->max_mem_alloc_size());
         copy_ptr = &val_sizet;
         size_ret = sizeof(val_sizet);
         break;