feat: Added compute shader support for vulkan

Vulkan graphics backend has been modified to support compute shaders,
additional modifications were also made to the shader compiler so that correct
glsl compute shaders can be generated.

Author: johang88

sources/engine/Stride.Graphics/Vulkan/CommandList.Vulkan.cs

@@ -130,9 +130,9 @@ private unsafe void FlushInternal(bool wait)
 
             if (activePipeline != null)
             {
-                vkCmdBindPipeline(currentCommandList.NativeCommandBuffer, VkPipelineBindPoint.Graphics, activePipeline.NativePipeline);
+                vkCmdBindPipeline(currentCommandList.NativeCommandBuffer, activePipeline.IsCompute ? VkPipelineBindPoint.Compute : VkPipelineBindPoint.Graphics, activePipeline.NativePipeline);
                 var descriptorSetCopy = descriptorSet;
-                vkCmdBindDescriptorSets(currentCommandList.NativeCommandBuffer, VkPipelineBindPoint.Graphics, activePipeline.NativeLayout, firstSet: 0, descriptorSetCount: 1, &descriptorSetCopy, dynamicOffsetCount: 0, dynamicOffsets: null);
+                vkCmdBindDescriptorSets(currentCommandList.NativeCommandBuffer, activePipeline.IsCompute ? VkPipelineBindPoint.Compute : VkPipelineBindPoint.Graphics, activePipeline.NativeLayout, firstSet: 0, descriptorSetCount: 1, &descriptorSetCopy, dynamicOffsetCount: 0, dynamicOffsets: null);
             }
             SetRenderTargetsImpl(depthStencilBuffer, renderTargetCount, renderTargets);
         }
@@ -249,7 +249,11 @@ private unsafe void PrepareDraw()
 
             // Lazily set the render pass and frame buffer
             EnsureRenderPass();
+            BindDescriptorSets();
+        }
 
+        private unsafe void BindDescriptorSets()
+        {
             // Keep track of descriptor pool usage
             bool isPoolExhausted = ++allocatedSetCount > GraphicsDevice.MaxDescriptorSetCount;
             for (int i = 0; i < DescriptorSetLayout.DescriptorTypeCount; i++)
@@ -328,18 +332,28 @@ private unsafe void PrepareDraw()
                     sType = VkStructureType.WriteDescriptorSet,
                     descriptorType = mapping.DescriptorType,
                     dstSet = localDescriptorSet,
-                    dstBinding = (uint) mapping.DestinationBinding,
+                    dstBinding = (uint)mapping.DestinationBinding,
                     dstArrayElement = 0,
                     descriptorCount = 1
                 };
 
                 switch (mapping.DescriptorType)
                 {
                     case VkDescriptorType.SampledImage:
-                        var texture = heapObject.Value as Texture;
-                        descriptorData->ImageInfo = new VkDescriptorImageInfo { imageView = texture?.NativeImageView ?? GraphicsDevice.EmptyTexture.NativeImageView, imageLayout = VkImageLayout.ShaderReadOnlyOptimal };
-                        write->pImageInfo = &descriptorData->ImageInfo;
-                        break;
+                        {
+                            var texture = heapObject.Value as Texture;
+                            descriptorData->ImageInfo = new VkDescriptorImageInfo { imageView = texture?.NativeImageView ?? GraphicsDevice.EmptyTexture.NativeImageView, imageLayout = VkImageLayout.ShaderReadOnlyOptimal };
+                            write->pImageInfo = &descriptorData->ImageInfo;
+                            break;
+                        }
+
+                    case VkDescriptorType.StorageImage:
+                        {
+                            var texture = heapObject.Value as Texture;
+                            descriptorData->ImageInfo = new VkDescriptorImageInfo { imageView = texture?.NativeImageView ?? GraphicsDevice.EmptyTexture.NativeImageView, imageLayout = VkImageLayout.General };
+                            write->pImageInfo = &descriptorData->ImageInfo;
+                            break;
+                        }
 
                     case VkDescriptorType.Sampler:
                         var samplerState = heapObject.Value as SamplerState;
@@ -349,7 +363,7 @@ private unsafe void PrepareDraw()
 
                     case VkDescriptorType.UniformBuffer:
                         var buffer = heapObject.Value as Buffer;
-                        descriptorData->BufferInfo = new VkDescriptorBufferInfo { buffer = buffer?.NativeBuffer ?? VkBuffer.Null, offset = (ulong) heapObject.Offset, range = (ulong) heapObject.Size };
+                        descriptorData->BufferInfo = new VkDescriptorBufferInfo { buffer = buffer?.NativeBuffer ?? VkBuffer.Null, offset = (ulong)heapObject.Offset, range = (ulong)heapObject.Size };
                         write->pBufferInfo = &descriptorData->BufferInfo;
                         break;
 
@@ -364,9 +378,9 @@ private unsafe void PrepareDraw()
                 }
             }
 
-            vkUpdateDescriptorSets(GraphicsDevice.NativeDevice, (uint) bindingCount, writes, descriptorCopyCount: 0, descriptorCopies: null);
+            vkUpdateDescriptorSets(GraphicsDevice.NativeDevice, (uint)bindingCount, writes, descriptorCopyCount: 0, descriptorCopies: null);
 #endif
-            vkCmdBindDescriptorSets(currentCommandList.NativeCommandBuffer, VkPipelineBindPoint.Graphics, activePipeline.NativeLayout, firstSet: 0, descriptorSetCount: 1, &localDescriptorSet, dynamicOffsetCount: 0, dynamicOffsets: null);
+            vkCmdBindDescriptorSets(currentCommandList.NativeCommandBuffer, activePipeline.IsCompute ? VkPipelineBindPoint.Compute : VkPipelineBindPoint.Graphics, activePipeline.NativeLayout, firstSet: 0, descriptorSetCount: 1, &localDescriptorSet, dynamicOffsetCount: 0, dynamicOffsets: null);
         }
 
         private readonly FastList<VkCopyDescriptorSet> copies = new();
@@ -390,7 +404,7 @@ public void SetPipelineState(PipelineState pipelineState)
 
             activePipeline = pipelineState;
 
-            vkCmdBindPipeline(currentCommandList.NativeCommandBuffer, VkPipelineBindPoint.Graphics, pipelineState.NativePipeline);
+            vkCmdBindPipeline(currentCommandList.NativeCommandBuffer, activePipeline.IsCompute ? VkPipelineBindPoint.Compute : VkPipelineBindPoint.Graphics, pipelineState.NativePipeline);
         }
 
         public unsafe void SetVertexBuffer(int index, Buffer buffer, int offset, int stride)
@@ -446,7 +460,7 @@ public unsafe void ResourceBarrierTransition(GraphicsResource resource, Graphics
                     case GraphicsResourceState.PixelShaderResource:
                         texture.NativeLayout = VkImageLayout.ShaderReadOnlyOptimal;
                         texture.NativeAccessMask = VkAccessFlags.ShaderRead;
-                        texture.NativePipelineStageMask = VkPipelineStageFlags.FragmentShader;
+                        texture.NativePipelineStageMask = VkPipelineStageFlags.FragmentShader | VkPipelineStageFlags.ComputeShader; // TODO: Not sure why I did this can probably double check ...
                         break;
                     case GraphicsResourceState.GenericRead:
                         texture.NativeLayout = VkImageLayout.General;
@@ -503,6 +517,9 @@ public void SetDescriptorSets(int index, DescriptorSet[] descriptorSets)
         /// <inheritdoc />
         public void Dispatch(int threadCountX, int threadCountY, int threadCountZ)
         {
+            CleanupRenderPass();
+            BindDescriptorSets();
+            vkCmdDispatch(currentCommandList.NativeCommandBuffer, (uint)threadCountX, (uint)threadCountY, (uint)threadCountZ);
         }
 
         /// <summary>
@@ -512,6 +529,9 @@ public void Dispatch(int threadCountX, int threadCountY, int threadCountZ)
         /// <param name="offsetInBytes">The offset information bytes.</param>
         public void Dispatch(Buffer indirectBuffer, int offsetInBytes)
         {
+            CleanupRenderPass();
+            BindDescriptorSets();
+            vkCmdDispatchIndirect(currentCommandList.NativeCommandBuffer, indirectBuffer.NativeBuffer, (ulong)offsetInBytes);
         }
 
         /// <summary>

sources/engine/Stride.Graphics/Vulkan/GraphicsDevice.Vulkan.cs

@@ -52,9 +52,9 @@ public partial class GraphicsDevice
             256, // Sampler
             0, // CombinedImageSampler
             512, // SampledImage
-            0, // StorageImage
+            32, // StorageImage
             64, // UniformTexelBuffer
-            0, // StorageTexelBuffer
+            32, // StorageTexelBuffer
             512, // UniformBuffer
             0, // StorageBuffer
             0, // UniformBufferDynamic
@@ -287,6 +287,12 @@ private unsafe void InitializePlatformDevice(GraphicsProfile[] graphicsProfiles,
                 depthClamp = true,
             };
 
+            if (graphicsProfiles.Any(x => x >= GraphicsProfile.Level_11_0))
+            {
+                enabledFeature.shaderStorageImageReadWithoutFormat = true;
+                enabledFeature.shaderStorageImageWriteWithoutFormat = true;
+            }
+
             var extensionProperties = vkEnumerateDeviceExtensionProperties(NativePhysicalDevice);
             var availableExtensionNames = new List<string>();
             var desiredExtensionNames = new List<string>();