Merge pull request #2994 from GeniusVentures/fix-vulkan-devices

Fixed Vulkan devices detection
alibaba · Aug 20, 2024 · e6753f5 · e6753f5
2 parents d88c469 + 5ec87b6
commit e6753f5
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/source/backend/vulkan/component/VulkanDevice.cpp b/source/backend/vulkan/component/VulkanDevice.cpp
@@ -25,9 +25,8 @@ VulkanDevice::VulkanDevice(std::shared_ptr<VulkanInstance> instance, const std::
     uint32_t gpuCount = 0;
     CALL_VK(mInstance->enumeratePhysicalDevices(gpuCount, nullptr));
     MNN_ASSERT(0 != gpuCount);
-    VkPhysicalDevice tmpGpus[1] = {nullptr};
-    gpuCount = 1;
-    CALL_VK(mInstance->enumeratePhysicalDevices(gpuCount, tmpGpus));
+    std::vector<VkPhysicalDevice> tmpGpus(gpuCount);
+    CALL_VK(mInstance->enumeratePhysicalDevices(gpuCount, tmpGpus.data()));
     MNN_ASSERT(nullptr != tmpGpus[0]);
     mPhysicalDevice = tmpGpus[0];