xref: /llvm-project/llvm/test/CodeGen/AMDGPU/promote-alloca-lifetime.ll (revision f0415f2a456d54daaa231c228d2c9f4ef2ce9b89)
1; RUN: opt -S -mtriple=amdgcn-unknown-amdhsa -passes=amdgpu-promote-alloca %s | FileCheck -check-prefix=OPT %s
2
3target datalayout = "e-p:64:64-p1:64:64-p2:32:32-p3:32:32-p4:64:64-p5:32:32-p6:32:32-p7:160:256:256:32-p8:128:128-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
5declare void @llvm.lifetime.start.p5(i64, ptr addrspace(5) nocapture) #0
6declare void @llvm.lifetime.end.p5(i64, ptr 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, ptr addrspace(3)
12define amdgpu_kernel void @use_lifetime_promotable_lds(ptr addrspace(1) %arg) #2 {
13bb:
14  %tmp = alloca i32, align 4, addrspace(5)
15  call void @llvm.lifetime.start.p5(i64 4, ptr addrspace(5) %tmp)
16  %tmp2 = getelementptr inbounds i32, ptr addrspace(1) %arg, i64 1
17  %tmp3 = load i32, ptr addrspace(1) %tmp2
18  store i32 %tmp3, ptr addrspace(5) %tmp
19  call void @llvm.lifetime.end.p5(i64 4, ptr addrspace(5) %tmp)
20  ret void
21}
22
23; After handling the alloca, the lifetime was erased. This was the
24; next iterator to be checked as an alloca, crashing.
25
26; OPT-LABEL: @iterator_erased_lifetime(
27; OPT-NOT: alloca i8
28define amdgpu_kernel void @iterator_erased_lifetime() {
29entry:
30  %alloca = alloca i8, align 1, addrspace(5)
31  call void @llvm.lifetime.start.p5(i64 1, ptr addrspace(5) %alloca)
32  ret void
33}
34
35attributes #0 = { argmemonly nounwind }
36attributes #1 = { nounwind readnone }
37attributes #2 = { nounwind }
38