From 8bea7198792206f283a652bafe5b73686490ce01 Mon Sep 17 00:00:00 2001
From: Aaron Miller <apage43@ninjawhale.com>
Date: Wed, 13 Sep 2023 09:51:40 -0700
Subject: [PATCH] vulkan: disambiguate gpus with the same name

---
 ggml-vulkan.cpp | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/ggml-vulkan.cpp b/ggml-vulkan.cpp
index c7bb3ed2b..378f1d6e6 100644
--- a/ggml-vulkan.cpp
+++ b/ggml-vulkan.cpp
@@ -145,6 +145,8 @@ std::vector<ggml_vk_device> ggml_vk_available_devices(size_t memoryRequired) {
     if (deviceCount == 0)
         return results;
 
+    std::unordered_map<std::string, size_t> count_by_name;
+
     for (uint32_t i = 0; i < deviceCount; i++) {
         VkPhysicalDeviceProperties properties = physicalDevices.at(i).getProperties();
         VkPhysicalDeviceMemoryProperties memoryProperties = physicalDevices.at(i).getMemoryProperties();
@@ -173,6 +175,10 @@ std::vector<ggml_vk_device> ggml_vk_available_devices(size_t memoryRequired) {
         d.type = properties.deviceType;
         d.heapSize = heapSize;
         d.name = properties.deviceName;
+        size_t n_idx = ++count_by_name[d.name];
+        if (n_idx > 1) {
+            d.name += " (" + std::to_string(n_idx) + ")";
+        }
         d.vendor = ggml_vk_getVendorName(properties.vendorID);
         results.push_back(d);
     }