1; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: -p --check-globals
2; RUN: opt -S -mtriple=amdgcn-- -amdgpu-lower-module-lds < %s | FileCheck %s
3; RUN: opt -S -mtriple=amdgcn-- -passes=amdgpu-lower-module-lds < %s | FileCheck %s
4
5@lds.1 = internal unnamed_addr addrspace(3) global [2 x i8] undef, align 1
6
7; CHECK: %llvm.amdgcn.kernel.k0.lds.t = type { [2 x i8] }
8; CHECK: %llvm.amdgcn.kernel.k1.lds.t = type { [2 x i8] }
9; CHECK: %llvm.amdgcn.kernel.k2.lds.t = type { i32 }
10; CHECK: %llvm.amdgcn.kernel.k3.lds.t = type { [32 x i8] }
11; CHECK: %llvm.amdgcn.kernel.k4.lds.t = type { [2 x i8] }
12
13; Use constant from different kernels
14;.
15; CHECK: @llvm.amdgcn.kernel.k0.lds = internal addrspace(3) global %llvm.amdgcn.kernel.k0.lds.t undef, align 2
16; CHECK: @llvm.amdgcn.kernel.k1.lds = internal addrspace(3) global %llvm.amdgcn.kernel.k1.lds.t undef, align 2
17; CHECK: @llvm.amdgcn.kernel.k2.lds = internal addrspace(3) global %llvm.amdgcn.kernel.k2.lds.t undef, align 4
18; CHECK: @llvm.amdgcn.kernel.k3.lds = internal addrspace(3) global %llvm.amdgcn.kernel.k3.lds.t undef, align 16
19; CHECK: @llvm.amdgcn.kernel.k4.lds = internal addrspace(3) global %llvm.amdgcn.kernel.k4.lds.t undef, align 2
20;.
21define amdgpu_kernel void @k0(i64 %x) {
22; CHECK-LABEL: @k0(
23; CHECK-NEXT:    %1 = getelementptr inbounds [2 x i8], [2 x i8] addrspace(3)* getelementptr inbounds (%llvm.amdgcn.kernel.k0.lds.t, %llvm.amdgcn.kernel.k0.lds.t addrspace(3)* @llvm.amdgcn.kernel.k0.lds, i32 0, i32 0), i32 0, i32 0
24; CHECK-NEXT:    %2 = addrspacecast i8 addrspace(3)* %1 to i8*
25; CHECK-NEXT:    %ptr = getelementptr inbounds i8, i8* %2, i64 %x
26; CHECK-NEXT:    store i8 1, i8* %ptr, align 1
27; CHECK-NEXT:    ret void
28;
29  %ptr = getelementptr inbounds i8, i8* addrspacecast ([2 x i8] addrspace(3)* @lds.1 to i8*), i64 %x
30  store i8 1, i8 addrspace(0)* %ptr, align 1
31  ret void
32}
33
34define amdgpu_kernel void @k1(i64 %x) {
35; CHECK-LABEL: @k1(
36; CHECK-NEXT:    %1 = getelementptr inbounds [2 x i8], [2 x i8] addrspace(3)* getelementptr inbounds (%llvm.amdgcn.kernel.k1.lds.t, %llvm.amdgcn.kernel.k1.lds.t addrspace(3)* @llvm.amdgcn.kernel.k1.lds, i32 0, i32 0), i32 0, i32 0
37; CHECK-NEXT:    %2 = addrspacecast i8 addrspace(3)* %1 to i8*
38; CHECK-NEXT:    %ptr = getelementptr inbounds i8, i8* %2, i64 %x
39; CHECK-NEXT:    store i8 1, i8* %ptr, align 1
40; CHECK-NEXT:    ret void
41;
42  %ptr = getelementptr inbounds i8, i8* addrspacecast ([2 x i8] addrspace(3)* @lds.1 to i8*), i64 %x
43  store i8 1, i8 addrspace(0)* %ptr, align 1
44  ret void
45}
46
47@lds.2 = internal unnamed_addr addrspace(3) global i32 undef, align 4
48
49; Use constant twice from the same kernel
50define amdgpu_kernel void @k2(i64 %x) {
51; CHECK-LABEL: @k2(
52; 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)*
53; CHECK-NEXT:    store i8 1, i8 addrspace(3)* %ptr1, align 4
54; 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)*
55; CHECK-NEXT:    store i8 2, i8 addrspace(3)* %ptr2, align 4
56; CHECK-NEXT:    ret void
57;
58  %ptr1 = bitcast i32 addrspace(3)* @lds.2 to i8 addrspace(3)*
59  store i8 1, i8 addrspace(3)* %ptr1, align 4
60  %ptr2 = bitcast i32 addrspace(3)* @lds.2 to i8 addrspace(3)*
61  store i8 2, i8 addrspace(3)* %ptr2, align 4
62  ret void
63}
64
65@lds.3 = internal unnamed_addr addrspace(3) global [32 x i8] undef, align 1
66
67; Use constant twice from the same kernel but a different other constant.
68define amdgpu_kernel void @k3(i64 %x) {
69; CHECK-LABEL: @k3(
70; CHECK-NEXT:    %1 = getelementptr inbounds [32 x i8], [32 x 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 0, i32 16
71; CHECK-NEXT:    %2 = bitcast i8 addrspace(3)* %1 to i64 addrspace(3)*
72; CHECK-NEXT:    %ptr1 = addrspacecast i64 addrspace(3)* %2 to i64*
73; CHECK-NEXT:    store i64 1, i64* %ptr1, align 1
74; CHECK-NEXT:    %3 = getelementptr inbounds [32 x i8], [32 x 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 0, i32 24
75; CHECK-NEXT:    %4 = bitcast i8 addrspace(3)* %3 to i64 addrspace(3)*
76; CHECK-NEXT:    %ptr2 = addrspacecast i64 addrspace(3)* %4 to i64*
77; CHECK-NEXT:    store i64 2, i64* %ptr2, align 1
78; CHECK-NEXT:    ret void
79;
80  %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*
81  store i64 1, i64* %ptr1, align 1
82  %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*
83  store i64 2, i64* %ptr2, align 1
84  ret void
85}
86
87; @lds.1 is used from constant expressions in different kernels.
88define amdgpu_kernel void @k4(i64 %x) {
89; CHECK-LABEL: @k4(
90; CHECK-NEXT:    %1 = getelementptr inbounds [2 x i8], [2 x i8] addrspace(3)* getelementptr inbounds (%llvm.amdgcn.kernel.k4.lds.t, %llvm.amdgcn.kernel.k4.lds.t addrspace(3)* @llvm.amdgcn.kernel.k4.lds, i32 0, i32 0), i32 0, i32 0
91; CHECK-NEXT:    %2 = addrspacecast i8 addrspace(3)* %1 to i8*
92; CHECK-NEXT:    %ptr = getelementptr inbounds i8, i8* %2, i64 %x
93; CHECK-NEXT:    store i8 1, i8* %ptr, align 1
94; CHECK-NEXT:    ret void
95;
96  %ptr = getelementptr inbounds i8, i8* addrspacecast ([2 x i8] addrspace(3)* @lds.1 to i8*), i64 %x
97  store i8 1, i8 addrspace(0)* %ptr, align 1
98  ret void
99}
100