From d35dfc5a3ff48331473ca42b97f3a8cba7d4ad9b Mon Sep 17 00:00:00 2001
From: Robin Voetter <robin@voetter.nl>
Date: Sat, 2 Nov 2024 18:54:34 +0100
Subject: [PATCH] add storage_buffer address space

---
 lib/std/Target.zig    | 2 +-
 lib/std/builtin.zig   | 1 +
 src/Sema.zig          | 2 +-
 src/codegen/spirv.zig | 1 +
 src/target.zig        | 2 +-
 5 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/lib/std/Target.zig b/lib/std/Target.zig
index 298da21187b055c0c755e6ace61d3ec1f61bf8ac..228300411504cd40cdf3209c1fa5521f3ff4237c 100644
--- a/lib/std/Target.zig
+++ b/lib/std/Target.zig
@@ -1573,7 +1573,7 @@ pub const Cpu = struct {
                 .fs, .gs, .ss => arch == .x86_64 or arch == .x86,
                 .global, .constant, .local, .shared => is_gpu,
                 .param => is_nvptx,
-                .input, .output, .uniform, .push_constant => is_spirv,
+                .input, .output, .uniform, .push_constant, .storage_buffer => is_spirv,
                 // TODO this should also check how many flash banks the cpu has
                 .flash, .flash1, .flash2, .flash3, .flash4, .flash5 => arch == .avr,
 
diff --git a/lib/std/builtin.zig b/lib/std/builtin.zig
index 02948d660206fa105f3aa4f55ef5692a2bb144f9..ed18965e50efaaf0c24826716623a5dd5a6656ab 100644
--- a/lib/std/builtin.zig
+++ b/lib/std/builtin.zig
@@ -515,6 +515,7 @@ pub const AddressSpace = enum(u5) {
     output,
     uniform,
     push_constant,
+    storage_buffer,
 
     // AVR address spaces.
     flash,
diff --git a/src/Sema.zig b/src/Sema.zig
index 3082cdd3d860619b7166c18e2c7927987f908e12..30f033b0a1926093d7822af88f50b10b7c37d8bf 100644
--- a/src/Sema.zig
+++ b/src/Sema.zig
@@ -37852,7 +37852,7 @@ pub fn analyzeAsAddressSpace(
         .gs, .fs, .ss => (arch == .x86 or arch == .x86_64) and ctx == .pointer,
         // TODO: check that .shared and .local are left uninitialized
         .param => is_nv,
-        .input, .output, .uniform, .push_constant => is_spirv,
+        .input, .output, .uniform, .push_constant, .storage_buffer => is_spirv,
         .global, .shared, .local => is_gpu,
         .constant => is_gpu and (ctx == .constant),
         // TODO this should also check how many flash banks the cpu has
diff --git a/src/codegen/spirv.zig b/src/codegen/spirv.zig
index baa37277e86578adb36dbb4574df03e7a9e4a854..9197a0959fe0d71f9309b04f33bccd41d86f5caf 100644
--- a/src/codegen/spirv.zig
+++ b/src/codegen/spirv.zig
@@ -1893,6 +1893,7 @@ const NavGen = struct {
             .input => .Input,
             .output => .Output,
             .uniform => .Uniform,
+            .storage_buffer => .StorageBuffer,
             .gs,
             .fs,
             .ss,
diff --git a/src/target.zig b/src/target.zig
index aba1fffcbc228d97d19ee9731e70e83542c52a9b..94bb842b12c4bf7e5b68db2f856783df4aebc489 100644
--- a/src/target.zig
+++ b/src/target.zig
@@ -458,7 +458,7 @@ pub fn arePointersLogical(target: std.Target, as: AddressSpace) bool {
         .global => false,
         // TODO: Allowed with VK_KHR_variable_pointers.
         .shared => true,
-        .constant, .local, .input, .output, .uniform, .push_constant => true,
+        .constant, .local, .input, .output, .uniform, .push_constant, .storage_buffer => true,
         else => unreachable,
     };
 }
-- 
2.54.0