llvm/test/CodeGen/AMDGPU/lower-kernel-and-module-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.8.align.8 = internal unnamed_addr addrspace(3) global [8 x i8] poison, align 8
   8 @lds.size.16.align.16 = internal unnamed_addr addrspace(3) global [16 x i8] poison, align 16
   9
  10 ; CHECK: %llvm.amdgcn.module.lds.t = type { [8 x i8], [1 x i8] }
  11 ; CHECK: %llvm.amdgcn.kernel.k0.lds.t = type { [16 x i8], [4 x i8], [2 x i8], [1 x i8] }
  12 ; CHECK: %llvm.amdgcn.kernel.k1.lds.t = type { [16 x i8], [4 x i8], [2 x i8] }
  13 ; CHECK: %llvm.amdgcn.kernel.k2.lds.t = type { [2 x i8] }
  14 ; CHECK: %llvm.amdgcn.kernel.k3.lds.t = type { [4 x i8] }
  15
  16 ;.
  17 ; CHECK: @llvm.amdgcn.module.lds = internal addrspace(3) global %llvm.amdgcn.module.lds.t poison, align 8, !absolute_symbol !0
  18 ; CHECK: @llvm.compiler.used = appending addrspace(1) global [1 x ptr] [ptr addrspacecast (ptr addrspace(3) @llvm.amdgcn.module.lds to ptr)], section "llvm.metadata"
  19 ; CHECK: @llvm.amdgcn.kernel.k0.lds = internal addrspace(3) global %llvm.amdgcn.kernel.k0.lds.t poison, align 16, !absolute_symbol !0
  20 ; CHECK: @llvm.amdgcn.kernel.k1.lds = internal addrspace(3) global %llvm.amdgcn.kernel.k1.lds.t poison, align 16, !absolute_symbol !0
  21 ; CHECK: @llvm.amdgcn.kernel.k2.lds = internal addrspace(3) global %llvm.amdgcn.kernel.k2.lds.t poison, align 2, !absolute_symbol !0
  22 ; CHECK: @llvm.amdgcn.kernel.k3.lds = internal addrspace(3) global %llvm.amdgcn.kernel.k3.lds.t poison, align 4, !absolute_symbol !0
  23 ;.
  24 define amdgpu_kernel void @k0() #0 {
  25 ; CHECK-LABEL: @k0() #0
  26 ; 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
  27 ; 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
  28 ; 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
  29 ; CHECK-NEXT: store i8 16, ptr addrspace(3) @llvm.amdgcn.kernel.k0.lds, align 16, !alias.scope !12, !noalias !13
  30 ; CHECK-NEXT:    ret void
  31   store i8 1, ptr addrspace(3) @lds.size.1.align.1, align 1
  32
  33   store i8 2, ptr addrspace(3) @lds.size.2.align.2, align 2
  34
  35   store i8 4, ptr addrspace(3) @lds.size.4.align.4, align 4
  36
  37   store i8 16, ptr addrspace(3) @lds.size.16.align.16, align 16
  38
  39   ret void
  40 }
  41
  42 define amdgpu_kernel void @k1() #0 {
  43 ; CHECK-LABEL: @k1() #1
  44 ; 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
  45 ; 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
  46 ; CHECK-NEXT: store i8 16, ptr addrspace(3) @llvm.amdgcn.kernel.k1.lds, align 16, !alias.scope !22, !noalias !23
  47 ; CHECK-NEXT:    ret void
  48 ;
  49   store i8 2, ptr addrspace(3) @lds.size.2.align.2, align 2
  50
  51   store i8 4, ptr addrspace(3) @lds.size.4.align.4, align 4
  52
  53   store i8 16, ptr addrspace(3) @lds.size.16.align.16, align 16
  54
  55   ret void
  56 }
  57
  58 define amdgpu_kernel void @k2() #0 {
  59 ; CHECK-LABEL: @k2() #2
  60 ; CHECK-NEXT:    store i8 2, ptr addrspace(3) @llvm.amdgcn.kernel.k2.lds, align 2
  61 ; CHECK-NEXT:    ret void
  62 ;
  63   store i8 2, ptr addrspace(3) @lds.size.2.align.2, align 2
  64
  65   ret void
  66 }
  67
  68 define amdgpu_kernel void @k3() #0 {
  69 ; CHECK-LABEL: @k3() #3
  70 ; CHECK-NEXT:    store i8 4, ptr addrspace(3) @llvm.amdgcn.kernel.k3.lds, align 4
  71 ; CHECK-NEXT:    ret void
  72 ;
  73   store i8 4, ptr addrspace(3) @lds.size.4.align.4, align 4
  74
  75   ret void
  76 }
  77
  78 ; CHECK-LABEL: @calls_f0() #4
  79 define amdgpu_kernel void @calls_f0() {
  80   call void @f0()
  81   ret void
  82 }
  83
  84 define void @f0() {
  85 ; CHECK-LABEL: define void @f0()
  86 ; CHECK-NEXT: store i8 1, ptr addrspace(3) getelementptr inbounds (%llvm.amdgcn.module.lds.t, ptr addrspace(3) @llvm.amdgcn.module.lds, i32 0, i32 1), align 8, !noalias !24
  87 ; CHECK-NEXT: store i8 8, ptr addrspace(3) @llvm.amdgcn.module.lds, align 8, !noalias !24
  88 ; CHECK-NEXT: ret void
  89   store i8 1, ptr addrspace(3) @lds.size.1.align.1, align 1
  90
  91   store i8 8, ptr addrspace(3) @lds.size.8.align.8, align 4
  92
  93   ret void
  94 }
  95
  96 ; CHECK: attributes #0 = { "amdgpu-lds-size"="23" }
  97 ; CHECK: attributes #1 = { "amdgpu-lds-size"="22" }
  98 ; CHECK: attributes #2 = { "amdgpu-lds-size"="2" }
  99 ; CHECK: attributes #3 = { "amdgpu-lds-size"="4" }
 100 ; CHECK: attributes #4 = { "amdgpu-lds-size"="9" }
 101
 102 ; CHECK: !0 = !{i32 0, i32 1}