llvm/test/CodeGen/AMDGPU/promote-alloca-lifetime.ll

   1 ; RUN: opt -S -mtriple=amdgcn-unknown-amdhsa -amdgpu-promote-alloca %s | FileCheck -check-prefix=OPT %s
   2
   3 target datalayout = "e-p:64:64-p1:64:64-p2:32:32-p3:32:32-p4:64:64-p5:32:32-p6:32:32-i64:64-v16:16-v24:32-v32:32-v48:64-v96:128-v192:256-v256:256-v512:512-v1024:1024-v2048:2048-n32:64-S32-A5"
   4
   5 declare void @llvm.lifetime.start.p5i8(i64, i8 addrspace(5)* nocapture) #0
   6 declare void @llvm.lifetime.end.p5i8(i64, i8 addrspace(5)* nocapture) #0
   7
   8 ; OPT-LABEL: @use_lifetime_promotable_lds(
   9 ; OPT-NOT: alloca i32
  10 ; OPT-NOT: llvm.lifetime
  11 ; OPT: store i32 %tmp3, i32 addrspace(3)*
  12 define amdgpu_kernel void @use_lifetime_promotable_lds(i32 addrspace(1)* %arg) #2 {
  13 bb:
  14   %tmp = alloca i32, align 4, addrspace(5)
  15   %tmp1 = bitcast i32 addrspace(5)* %tmp to i8 addrspace(5)*
  16   call void @llvm.lifetime.start.p5i8(i64 4, i8 addrspace(5)* %tmp1)
  17   %tmp2 = getelementptr inbounds i32, i32 addrspace(1)* %arg, i64 1
  18   %tmp3 = load i32, i32 addrspace(1)* %tmp2
  19   store i32 %tmp3, i32 addrspace(5)* %tmp
  20   call void @llvm.lifetime.end.p5i8(i64 4, i8 addrspace(5)* %tmp1)
  21   ret void
  22 }
  23
  24 ; After handling the alloca, the lifetime was erased. This was the
  25 ; next iterator to be checked as an alloca, crashing.
  26
  27 ; OPT-LABEL: @iterator_erased_lifetime(
  28 ; OPT-NOT: alloca i8
  29 define amdgpu_kernel void @iterator_erased_lifetime() {
  30 entry:
  31   %alloca = alloca i8, align 1, addrspace(5)
  32   call void @llvm.lifetime.start.p5i8(i64 1, i8 addrspace(5)* %alloca)
  33   ret void
  34 }
  35
  36 attributes #0 = { argmemonly nounwind }
  37 attributes #1 = { nounwind readnone }
  38 attributes #2 = { nounwind }