From 0d5e75dfaab7b85f6d0db0076017b3c35b8dc753 Mon Sep 17 00:00:00 2001 From: 0cc4m Date: Sun, 9 Nov 2025 15:51:16 +0100 Subject: [PATCH] vulkan: fix memory allocations --- ggml/src/ggml-vulkan/ggml-vulkan.cpp | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/ggml/src/ggml-vulkan/ggml-vulkan.cpp b/ggml/src/ggml-vulkan/ggml-vulkan.cpp index 46e098a7ffb11..7570febeaada3 100644 --- a/ggml/src/ggml-vulkan/ggml-vulkan.cpp +++ b/ggml/src/ggml-vulkan/ggml-vulkan.cpp @@ -2220,9 +2220,12 @@ static vk_buffer ggml_vk_create_buffer(vk_device& device, size_t size, const std } buf->memory_property_flags = req_flags; + bool done = false; + for (auto mtype_it = memory_type_indices.begin(); mtype_it != memory_type_indices.end(); mtype_it++) { try { buf->device_memory = device->device.allocateMemory({ mem_req.size, *mtype_it, &mem_flags_info }); + done = true; break; } catch (const vk::SystemError& e) { // loop and retry @@ -2233,6 +2236,10 @@ static vk_buffer ggml_vk_create_buffer(vk_device& device, size_t size, const std } } } + + if (done) { + break; + } } if (!buf->device_memory) {