Use dispatch params shared memory size when available

2 files changed, 12 insertions, 1 deletions

diff --git a/Ryujinx.Graphics.Shader/CodeGen/Glsl/Declarations.cs b/Ryujinx.Graphics.Shader/CodeGen/Glsl/Declarations.cs
index 1c3cf245..b96aa1ae 100644
--- a/Ryujinx.Graphics.Shader/CodeGen/Glsl/Declarations.cs
+++ b/Ryujinx.Graphics.Shader/CodeGen/Glsl/Declarations.cs
@@ -75,7 +75,16 @@ namespace Ryujinx.Graphics.Shader.CodeGen.Glsl
 
             if (context.Config.Stage == ShaderStage.Compute)
             {
-                string size = NumberFormatter.FormatInt(context.Config.Capabilities.MaximumComputeSharedMemorySize / 4);
+                string size;
+
+                if ((context.Config.Flags & TranslationFlags.Unspecialized) != 0)
+                {
+                    size = DefineNames.SharedMemorySize;
+                }
+                else
+                {
+                    size = NumberFormatter.FormatInt(context.Config.Capabilities.MaximumComputeSharedMemorySize / 4);
+                }
 
                 context.AppendLine($"shared uint {DefaultNames.SharedMemoryName}[{size}];");
                 context.AppendLine();
diff --git a/Ryujinx.Graphics.Shader/DefineNames.cs b/Ryujinx.Graphics.Shader/DefineNames.cs
index f55cbd0c..67e8e1ee 100644
--- a/Ryujinx.Graphics.Shader/DefineNames.cs
+++ b/Ryujinx.Graphics.Shader/DefineNames.cs
@@ -6,6 +6,8 @@ namespace Ryujinx.Graphics.Shader
 
         public const string OutQualifierPrefixName = "S_OUT_QUALIFIER";
 
+        public const string SharedMemorySize = "S_SHARED_MEMORY_SIZE";
+
         public const string LocalSizeX = "S_LOCAL_SIZE_X";
         public const string LocalSizeY = "S_LOCAL_SIZE_Y";
         public const string LocalSizeZ = "S_LOCAL_SIZE_Z";