1; RUN: opt -S -mtriple=amdgcn-- -amdgpu-lower-module-lds < %s | FileCheck %s
2; RUN: opt -S -mtriple=amdgcn-- -passes=amdgpu-lower-module-lds < %s | FileCheck %s
3
4; CHECK: %llvm.amdgcn.kernel.k2.lds.t = type { i32 }
5; CHECK-NOT: %llvm.amdgcn.kernel.k4.lds.t
6
7@lds.1 = internal unnamed_addr addrspace(3) global [2 x i8] undef, align 1
8
9; Use constant from different kernels
10;.
11; CHECK: @lds.1 = internal unnamed_addr addrspace(3) global [2 x i8] undef, align 1
12; CHECK: @llvm.amdgcn.kernel.k2.lds = internal addrspace(3) global %llvm.amdgcn.kernel.k2.lds.t undef, align 4
13;.
14define amdgpu_kernel void @k0(i64 %x) {
15; CHECK-LABEL: @k0(
16; CHECK-NEXT:    %ptr = getelementptr inbounds i8, i8* addrspacecast (i8 addrspace(3)* getelementptr inbounds ([2 x i8], [2 x i8] addrspace(3)* @lds.1, i32 0, i32 0) to i8*), i64 %x
17; CHECK-NEXT:    store i8 1, i8* %ptr, align 1
18; CHECK-NEXT:    ret void
19;
20  %ptr = getelementptr inbounds i8, i8* addrspacecast ([2 x i8] addrspace(3)* @lds.1 to i8*), i64 %x
21  store i8 1, i8 addrspace(0)* %ptr, align 1
22  ret void
23}
24
25define amdgpu_kernel void @k1(i64 %x) {
26; CHECK-LABEL: @k1(
27; CHECK-NEXT:    %ptr = getelementptr inbounds i8, i8* addrspacecast (i8 addrspace(3)* getelementptr inbounds ([2 x i8], [2 x i8] addrspace(3)* @lds.1, i32 0, i32 0) to i8*), i64 %x
28; CHECK-NEXT:    store i8 1, i8* %ptr, align 1
29; CHECK-NEXT:    ret void
30;
31  %ptr = getelementptr inbounds i8, i8* addrspacecast ([2 x i8] addrspace(3)* @lds.1 to i8*), i64 %x
32  store i8 1, i8 addrspace(0)* %ptr, align 1
33  ret void
34}
35
36@lds.2 = internal unnamed_addr addrspace(3) global i32 undef, align 4
37
38; Use constant twice from the same kernel
39define amdgpu_kernel void @k2(i64 %x) {
40; CHECK-LABEL: @k2(
41; CHECK-NEXT:    %ptr1 = bitcast i32 addrspace(3)* getelementptr inbounds (%llvm.amdgcn.kernel.k2.lds.t, %llvm.amdgcn.kernel.k2.lds.t addrspace(3)* @llvm.amdgcn.kernel.k2.lds, i32 0, i32 0) to i8 addrspace(3)*
42; CHECK-NEXT:    store i8 1, i8 addrspace(3)* %ptr1, align 4
43; CHECK-NEXT:    %ptr2 = bitcast i32 addrspace(3)* getelementptr inbounds (%llvm.amdgcn.kernel.k2.lds.t, %llvm.amdgcn.kernel.k2.lds.t addrspace(3)* @llvm.amdgcn.kernel.k2.lds, i32 0, i32 0) to i8 addrspace(3)*
44; CHECK-NEXT:    store i8 2, i8 addrspace(3)* %ptr2, align 4
45; CHECK-NEXT:    ret void
46;
47  %ptr1 = bitcast i32 addrspace(3)* @lds.2 to i8 addrspace(3)*
48  store i8 1, i8 addrspace(3)* %ptr1, align 4
49  %ptr2 = bitcast i32 addrspace(3)* @lds.2 to i8 addrspace(3)*
50  store i8 2, i8 addrspace(3)* %ptr2, align 4
51  ret void
52}
53
54@lds.3 = internal unnamed_addr addrspace(3) global [32 x i8] undef, align 1
55
56; Use constant twice from the same kernel but a different other constant.
57define amdgpu_kernel void @k3(i64 %x) {
58; CHECK-LABEL: @k3(
59; CHECK-NEXT:    %ptr1 = addrspacecast i64 addrspace(3)* bitcast (i8 addrspace(3)* getelementptr inbounds (%llvm.amdgcn.kernel.k3.lds.t, %llvm.amdgcn.kernel.k3.lds.t addrspace(3)* @llvm.amdgcn.kernel.k3.lds, i32 0, i32 0, i32 16) to i64 addrspace(3)*) to i64*
60; CHECK-NEXT:    store i64 1, i64* %ptr1, align 1
61; CHECK-NEXT:    %ptr2 = addrspacecast i64 addrspace(3)* bitcast (i8 addrspace(3)* getelementptr inbounds (%llvm.amdgcn.kernel.k3.lds.t, %llvm.amdgcn.kernel.k3.lds.t addrspace(3)* @llvm.amdgcn.kernel.k3.lds, i32 0, i32 0, i32 24) to i64 addrspace(3)*) to i64*
62; CHECK-NEXT:    store i64 2, i64* %ptr2, align 1
63;
64  %ptr1 = addrspacecast i64 addrspace(3)* bitcast (i8 addrspace(3)* getelementptr inbounds ([32 x i8], [32 x i8] addrspace(3)* @lds.3, i32 0, i32 16) to i64 addrspace(3)*) to i64*
65  store i64 1, i64* %ptr1, align 1
66  %ptr2 = addrspacecast i64 addrspace(3)* bitcast (i8 addrspace(3)* getelementptr inbounds ([32 x i8], [32 x i8] addrspace(3)* @lds.3, i32 0, i32 24) to i64 addrspace(3)*) to i64*
67  store i64 2, i64* %ptr2, align 1
68  ret void
69}
70
71; @lds.1 is used from constant expressions in different kernels.
72; Make sure we do not create a structure for it as we cannot handle it yet.
73define amdgpu_kernel void @k4(i64 %x) {
74; CHECK-LABEL: @k4(
75; CHECK-NEXT:    %ptr = getelementptr inbounds i8, i8* addrspacecast (i8 addrspace(3)* getelementptr inbounds ([2 x i8], [2 x i8] addrspace(3)* @lds.1, i32 0, i32 0) to i8*), i64 %x
76; CHECK-NEXT:    store i8 1, i8* %ptr, align 1
77; CHECK-NEXT:    ret void
78;
79  %ptr = getelementptr inbounds i8, i8* addrspacecast ([2 x i8] addrspace(3)* @lds.1 to i8*), i64 %x
80  store i8 1, i8 addrspace(0)* %ptr, align 1
81  ret void
82}
83