Use maximum shared memory size supported by hardware

2 files changed, 12 insertions, 6 deletions

diff --git a/Ryujinx.Graphics.Shader/CodeGen/Glsl/Declarations.cs b/Ryujinx.Graphics.Shader/CodeGen/Glsl/Declarations.cs
index 5a84ff49..1c3cf245 100644
--- a/Ryujinx.Graphics.Shader/CodeGen/Glsl/Declarations.cs
+++ b/Ryujinx.Graphics.Shader/CodeGen/Glsl/Declarations.cs
@@ -75,7 +75,9 @@ namespace Ryujinx.Graphics.Shader.CodeGen.Glsl
 
             if (context.Config.Stage == ShaderStage.Compute)
             {
-                context.AppendLine($"shared uint {DefaultNames.SharedMemoryName}[0x100];");
+                string size = NumberFormatter.FormatInt(context.Config.Capabilities.MaximumComputeSharedMemorySize / 4);
+
+                context.AppendLine($"shared uint {DefaultNames.SharedMemoryName}[{size}];");
                 context.AppendLine();
             }
 
diff --git a/Ryujinx.Graphics.Shader/ShaderCapabilities.cs b/Ryujinx.Graphics.Shader/ShaderCapabilities.cs
index b5b459a8..8e0c95e9 100644
--- a/Ryujinx.Graphics.Shader/ShaderCapabilities.cs
+++ b/Ryujinx.Graphics.Shader/ShaderCapabilities.cs
@@ -2,19 +2,23 @@ namespace Ryujinx.Graphics.Shader
 {
     public struct ShaderCapabilities
     {
-        private static readonly ShaderCapabilities _default = new ShaderCapabilities(32768, 16);
+        // Initialize with default values for Maxwell.
+        private static readonly ShaderCapabilities _default = new ShaderCapabilities(32768, 49152, 16);
 
         public static ShaderCapabilities Default => _default;
 
-        public int MaximumViewportDimensions    { get; }
-        public int StorageBufferOffsetAlignment { get; }
+        public int MaximumViewportDimensions      { get; }
+        public int MaximumComputeSharedMemorySize { get; }
+        public int StorageBufferOffsetAlignment   { get; }
 
         public ShaderCapabilities(
             int maximumViewportDimensions,
+            int maximumComputeSharedMemorySize,
             int storageBufferOffsetAlignment)
         {
-            MaximumViewportDimensions    = maximumViewportDimensions;
-            StorageBufferOffsetAlignment = storageBufferOffsetAlignment;
+            MaximumViewportDimensions      = maximumViewportDimensions;
+            MaximumComputeSharedMemorySize = maximumComputeSharedMemorySize;
+            StorageBufferOffsetAlignment   = storageBufferOffsetAlignment;
         }
     }
 }
\ No newline at end of file