llvm/test/CodeGen/AMDGPU/lower-kernel-lds.ll

   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 ;.
  17 define 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
  35 define 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
  55 define 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}