xref: /llvm-project/llvm/test/CodeGen/AMDGPU/lower-kernel-lds.ll (revision e39f6c1844fab59c638d8059a6cf139adb42279a)
1; RUN: opt -S -mtriple=amdgcn-- -amdgpu-lower-module-lds --amdgpu-lower-module-lds-strategy=module < %s | FileCheck %s
2; RUN: opt -S -mtriple=amdgcn-- -passes=amdgpu-lower-module-lds --amdgpu-lower-module-lds-strategy=module < %s | FileCheck %s
3
4@lds.size.1.align.1 = internal unnamed_addr addrspace(3) global [1 x i8] poison, align 1
5@lds.size.2.align.2 = internal unnamed_addr addrspace(3) global [2 x i8] poison, align 2
6@lds.size.4.align.4 = internal unnamed_addr addrspace(3) global [4 x i8] poison, align 4
7@lds.size.16.align.16 = internal unnamed_addr addrspace(3) global [16 x i8] poison, align 16
8
9; CHECK: %llvm.amdgcn.kernel.k0.lds.t = type { [16 x i8], [4 x i8], [2 x i8], [1 x i8] }
10; CHECK: %llvm.amdgcn.kernel.k1.lds.t = type { [16 x i8], [4 x i8], [2 x i8] }
11
12;.
13; CHECK: @lds.k2 = addrspace(3) global [1 x i8] poison, align 1
14; CHECK: @llvm.amdgcn.kernel.k0.lds = internal addrspace(3) global %llvm.amdgcn.kernel.k0.lds.t poison, align 16, !absolute_symbol !0
15; CHECK: @llvm.amdgcn.kernel.k1.lds = internal addrspace(3) global %llvm.amdgcn.kernel.k1.lds.t poison, align 16, !absolute_symbol !0
16;.
17define amdgpu_kernel void @k0() {
18; CHECK-LABEL: @k0(
19; CHECK-NEXT: store i8 1, ptr addrspace(3) getelementptr inbounds (%llvm.amdgcn.kernel.k0.lds.t, ptr addrspace(3) @llvm.amdgcn.kernel.k0.lds, i32 0, i32 3), align 2, !alias.scope !1, !noalias !4
20; CHECK-NEXT: store i8 2, ptr addrspace(3) getelementptr inbounds (%llvm.amdgcn.kernel.k0.lds.t, ptr addrspace(3) @llvm.amdgcn.kernel.k0.lds, i32 0, i32 2), align 4, !alias.scope !8, !noalias !9
21; CHECK-NEXT: store i8 4, ptr addrspace(3) getelementptr inbounds (%llvm.amdgcn.kernel.k0.lds.t, ptr addrspace(3) @llvm.amdgcn.kernel.k0.lds, i32 0, i32 1), align 16, !alias.scope !10, !noalias !11
22; CHECK-NEXT: store i8 16, ptr addrspace(3) @llvm.amdgcn.kernel.k0.lds, align 16, !alias.scope !12, !noalias !13
23; CHECK-NEXT: ret void
24  store i8 1, ptr addrspace(3) @lds.size.1.align.1, align 1
25
26  store i8 2, ptr addrspace(3) @lds.size.2.align.2, align 2
27
28  store i8 4, ptr addrspace(3) @lds.size.4.align.4, align 4
29
30  store i8 16, ptr addrspace(3) @lds.size.16.align.16, align 16
31
32  ret void
33}
34
35define amdgpu_kernel void @k1() {
36; CHECK-LABEL: @k1(
37; CHECK-NEXT: store i8 2, ptr addrspace(3) getelementptr inbounds (%llvm.amdgcn.kernel.k1.lds.t, ptr addrspace(3) @llvm.amdgcn.kernel.k1.lds, i32 0, i32 2), align 4, !alias.scope !14, !noalias !17
38; CHECK-NEXT: store i8 4, ptr addrspace(3) getelementptr inbounds (%llvm.amdgcn.kernel.k1.lds.t, ptr addrspace(3) @llvm.amdgcn.kernel.k1.lds, i32 0, i32 1), align 16, !alias.scope !20, !noalias !21
39; CHECK-NEXT: store i8 16, ptr addrspace(3) @llvm.amdgcn.kernel.k1.lds, align 16, !alias.scope !22, !noalias !23
40; CHECK-NEXT: ret void
41;
42  store i8 2, ptr addrspace(3) @lds.size.2.align.2, align 2
43
44  store i8 4, ptr addrspace(3) @lds.size.4.align.4, align 4
45
46  store i8 16, ptr addrspace(3) @lds.size.16.align.16, align 16
47
48  ret void
49}
50
51; Do not lower LDS for graphics shaders.
52
53@lds.k2 = addrspace(3) global [1 x i8] poison, align 1
54
55define amdgpu_ps void @k2() {
56; CHECK-LABEL: @k2(
57; CHECK-NEXT:    store i8 1, ptr addrspace(3) @lds.k2, align 1
58; CHECK-NEXT:    ret void
59;
60  store i8 1, ptr addrspace(3) @lds.k2, align 1
61
62  ret void
63}
64
65; CHECK: !0 = !{i32 0, i32 1}
66