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];