Vulkan external device

Author: MaxwellGengYF

include/luisa/backends/ext/vk_config_ext.h

@@ -8,11 +8,23 @@ namespace luisa::compute {
 
 class VulkanDeviceConfigExt : public DeviceConfigExt {
 public:
+    struct ExternalDevice {
+        VkInstance instance{};
+        VkPhysicalDevice physical_device{};
+        VkDevice device{};
+        VkQueue graphics_queue{};
+        VkQueue compute_queue{};
+        VkQueue copy_queue{};
+    };
     VulkanDeviceConfigExt() = default;
     ~VulkanDeviceConfigExt() = default;
+    [[nodiscard]] virtual ExternalDevice create_external_device() {
+        return {};
+    }
     [[nodiscard]] virtual bool enable_fallback() const {
         return false;
     }
+
     virtual VkCommandBuffer borrow_command_buffer(
         StreamTag stream_tag) noexcept { return nullptr; }
     virtual bool execute_command_buffer(VkCommandBuffer cmd_buffer) noexcept { return false; }

src/backends/vk/VulkanDevice.cpp

@@ -279,15 +279,17 @@ VkResult VulkanDevice::createLogicalDevice(VkPhysicalDeviceFeatures &enabledFeat
     }
 
     this->enabledFeatures = enabledFeatures;
+    if (!logicalDevice) {
+        VkResult result = vkCreateDevice(physicalDevice, &deviceCreateInfo, lc::vk::Device::alloc_callbacks(), &logicalDevice);
+        if (result != VK_SUCCESS) {
+            return result;
+        }
+
+        // Create a default command pool for graphics command buffers
 
-    VkResult result = vkCreateDevice(physicalDevice, &deviceCreateInfo, lc::vk::Device::alloc_callbacks(), &logicalDevice);
-    if (result != VK_SUCCESS) {
         return result;
     }
-
-    // Create a default command pool for graphics command buffers
-
-    return result;
+    return VK_SUCCESS;
 }
 
 bool VulkanDevice::extensionSupported(vstd::string_view extension) {