From e09465fc49d86cc4aa9338106862d3e059ae3303 Mon Sep 17 00:00:00 2001
From: Robin Voetter <robin@voetter.nl>
Date: Sun, 29 Aug 2021 03:18:23 +0200
Subject: [PATCH] Address Spaces: Chaining tests

---
 src/codegen/llvm/bindings.zig | 59 +++++++++++++++++++++++++++++++++++
 test/stage2/llvm.zig          | 52 ++++++++++++++++++++++++++++++
 2 files changed, 111 insertions(+)

diff --git a/src/codegen/llvm/bindings.zig b/src/codegen/llvm/bindings.zig
index db1dcd22f2..67b39784b1 100644
--- a/src/codegen/llvm/bindings.zig
+++ b/src/codegen/llvm/bindings.zig
@@ -973,3 +973,62 @@ pub const TypeKind = enum(c_int) {
     BFloat,
     X86_AMX,
 };
+
+pub const address_space = struct {
+    // See llvm/lib/Target/X86/X86.h
+    pub const x86_64 = x86;
+    pub const x86 = struct {
+        pub const gs = 256;
+        pub const fs = 257;
+        pub const ss = 258;
+
+        pub const ptr32_sptr = 270;
+        pub const ptr32_uptr = 271;
+        pub const ptr64 = 272;
+    };
+
+    // See llvm/lib/Target/AVR/AVR.h
+    pub const avr = struct {
+        pub const data_memory = 0;
+        pub const program_memory = 1;
+    };
+
+    // See llvm/lib/Target/NVPTX/NVPTX.h
+    pub const nvptx = struct {
+        pub const generic = 0;
+        pub const global = 1;
+        pub const constant = 2;
+        pub const shared = 3;
+        pub const param = 4;
+        pub const local = 5;
+    };
+
+    // See llvm/lib/Target/AMDGPU/AMDGPU.h
+    pub const amdgpu = struct {
+        pub const flat = 0;
+        pub const global = 1;
+        pub const region = 2;
+        pub const local = 3;
+        pub const constant = 4;
+        pub const constant_32bit = 6;
+        pub const buffer_fat_pointer = 7;
+        pub const param_d = 6;
+        pub const param_i = 7;
+        pub const constant_buffer_0 = 8;
+        pub const constant_buffer_1 = 9;
+        pub const constant_buffer_2 = 10;
+        pub const constant_buffer_3 = 11;
+        pub const constant_buffer_4 = 12;
+        pub const constant_buffer_5 = 13;
+        pub const constant_buffer_6 = 14;
+        pub const constant_buffer_7 = 15;
+        pub const constant_buffer_8 = 16;
+        pub const constant_buffer_9 = 17;
+        pub const constant_buffer_10 = 18;
+        pub const constant_buffer_11 = 19;
+        pub const constant_buffer_12 = 20;
+        pub const constant_buffer_13 = 21;
+        pub const constant_buffer_14 = 22;
+        pub const constant_buffer_15 = 23;
+    };
+};
diff --git a/test/stage2/llvm.zig b/test/stage2/llvm.zig
index b5ec47c5de..e61e7181c6 100644
--- a/test/stage2/llvm.zig
+++ b/test/stage2/llvm.zig
@@ -302,4 +302,56 @@ pub fn addCases(ctx: *TestContext) !void {
             \\pub export fn main() void { _ = entry; }
         );
     }
+
+    {
+        var case = ctx.exeUsingLlvmBackend("address spaces pointer access chaining: array pointer", linux_x64);
+        case.compiles(
+            \\fn entry(a: *addrspace(.gs) [1]i32) *addrspace(.gs) i32 {
+            \\    return &a[0];
+            \\}
+            \\pub export fn main() void { _ = entry; }
+        );
+    }
+
+    {
+        var case = ctx.exeUsingLlvmBackend("address spaces pointer access chaining: pointer to optional array", linux_x64);
+        case.compiles(
+            \\fn entry(a: *addrspace(.gs) ?[1]i32) *addrspace(.gs) i32 {
+            \\    return &a.*.?[0];
+            \\}
+            \\pub export fn main() void { _ = entry; }
+        );
+    }
+
+    {
+        var case = ctx.exeUsingLlvmBackend("address spaces pointer access chaining: struct pointer", linux_x64);
+        case.compiles(
+            \\const A = struct{ a: i32 };
+            \\fn entry(a: *addrspace(.gs) A) *addrspace(.gs) i32 {
+            \\    return &a.a;
+            \\}
+            \\pub export fn main() void { _ = entry; }
+        );
+    }
+
+    {
+        var case = ctx.exeUsingLlvmBackend("address spaces pointer access chaining: complex", linux_x64);
+        case.compiles(
+            \\const A = struct{ a: ?[1]i32 };
+            \\fn entry(a: *addrspace(.gs) [1]A) *addrspace(.gs) i32 {
+            \\    return &a[0].a.?[0];
+            \\}
+            \\pub export fn main() void { _ = entry; }
+        );
+    }
+
+    {
+        var case = ctx.exeUsingLlvmBackend("dereferencing through multiple pointers with address spaces", linux_x64);
+        case.compiles(
+            \\fn entry(a: *addrspace(.fs) *addrspace(.gs) *i32) *i32 {
+            \\    return a.*.*;
+            \\}
+            \\pub export fn main() void { _ = entry; }
+        );
+    }
 }