add preliminary vulkan bindless support (#337)

Author: skallweitNV

CMakeLists.txt

@@ -573,6 +573,7 @@ if(SLANG_RHI_ENABLE_VULKAN)
     target_sources(slang-rhi PRIVATE
         src/vulkan/vk-acceleration-structure.cpp
         src/vulkan/vk-api.cpp
+        src/vulkan/vk-bindless-descriptor-set.cpp
         src/vulkan/vk-buffer.cpp
         src/vulkan/vk-command.cpp
         src/vulkan/vk-constant-buffer-pool.cpp

src/vulkan/vk-acceleration-structure.cpp

@@ -36,6 +36,26 @@ DeviceAddress AccelerationStructureImpl::getDeviceAddress()
     return m_buffer->getDeviceAddress();
 }
 
+Result AccelerationStructureImpl::getDescriptorHandle(DescriptorHandle* outHandle)
+{
+    DeviceImpl* device = getDevice<DeviceImpl>();
+
+    if (!device->m_bindlessDescriptorSet)
+    {
+        return SLANG_E_NOT_AVAILABLE;
+    }
+
+    if (!m_descriptorHandle)
+    {
+        SLANG_RETURN_ON_FAIL(
+            device->m_bindlessDescriptorSet->allocAccelerationStructureHandle(this, &m_descriptorHandle)
+        );
+    }
+
+    *outHandle = m_descriptorHandle;
+    return SLANG_OK;
+}
+
 Result AccelerationStructureBuildDescConverter::convert(
     const AccelerationStructureBuildDesc& buildDesc,
     IDebugCallback* debugCallback

src/vulkan/vk-acceleration-structure.h

@@ -9,6 +9,7 @@ class AccelerationStructureImpl : public AccelerationStructure
 public:
     VkAccelerationStructureKHR m_vkHandle = VK_NULL_HANDLE;
     RefPtr<BufferImpl> m_buffer;
+    DescriptorHandle m_descriptorHandle;
 
 public:
     AccelerationStructureImpl(Device* device, const AccelerationStructureDesc& desc);
@@ -18,6 +19,7 @@ class AccelerationStructureImpl : public AccelerationStructure
     virtual SLANG_NO_THROW Result SLANG_MCALL getNativeHandle(NativeHandle* outHandle) override;
     virtual SLANG_NO_THROW AccelerationStructureHandle getHandle() override;
     virtual SLANG_NO_THROW DeviceAddress SLANG_MCALL getDeviceAddress() override;
+    virtual SLANG_NO_THROW Result SLANG_MCALL getDescriptorHandle(DescriptorHandle* outHandle) override;
 };
 
 struct AccelerationStructureBuildDescConverter

src/vulkan/vk-api.h

@@ -221,6 +221,7 @@ namespace rhi::vk {
     x(vkCmdDrawMeshTasksEXT) \
     x(vkConvertCooperativeVectorMatrixNV) \
     x(vkCmdConvertCooperativeVectorMatrixNV) \
+    x(vkGetDescriptorSetLayoutSupport) \
     /* */
 
 #define VK_API_ALL_GLOBAL_PROCS(x) \
@@ -373,6 +374,16 @@ struct VulkanExtendedFeatureProperties
     VkPhysicalDeviceCooperativeMatrixFeaturesKHR cooperativeMatrix1Features = {
         VK_STRUCTURE_TYPE_PHYSICAL_DEVICE_COOPERATIVE_MATRIX_FEATURES_KHR
     };
+
+    // Descriptor indexing features
+    VkPhysicalDeviceDescriptorIndexingFeatures descriptorIndexingFeatures = {
+        VK_STRUCTURE_TYPE_PHYSICAL_DEVICE_DESCRIPTOR_INDEXING_FEATURES
+    };
+
+    // Mutable descriptor type features
+    VkPhysicalDeviceMutableDescriptorTypeFeaturesEXT mutableDescriptorTypeFeatures = {
+        VK_STRUCTURE_TYPE_PHYSICAL_DEVICE_MUTABLE_DESCRIPTOR_TYPE_FEATURES_EXT
+    };
 };
 
 struct VulkanApi