xref: /llvm-project/llvm/test/Transforms/Coroutines/coro-frame-reuse-alloca-00.ll (revision 51d5d7bbae92493a5bfa7cc6b519de8a5bb32fdb) 
1; Check that we can handle spills of array allocas
2; RUN: opt < %s -passes='cgscc(coro-split<reuse-storage>),simplifycfg,early-cse' -S | FileCheck %s
3
4%struct.big_structure = type { [500 x i8] }
5declare void @consume(ptr)
6
7; Function Attrs: noinline optnone uwtable
8define ptr @f(i1 %cond) presplitcoroutine {
9entry:
10  %data = alloca %struct.big_structure, align 1
11  %data2 = alloca %struct.big_structure, align 1
12  %id = call token @llvm.coro.id(i32 0, ptr null, ptr null, ptr null)
13  %size = call i32 @llvm.coro.size.i32()
14  %alloc = call ptr @malloc(i32 %size)
15  %hdl = call ptr @llvm.coro.begin(token %id, ptr %alloc)
16  br i1 %cond, label %then, label %else
17
18then:
19  call void @llvm.lifetime.start.p0(i64 500, ptr nonnull %data)
20  call void @consume(ptr %data)
21  %suspend.value = call i8 @llvm.coro.suspend(token none, i1 false)
22  switch i8 %suspend.value, label %coro.ret [i8 0, label %resume
23                                             i8 1, label %cleanup1]
24
25resume:
26  call void @llvm.lifetime.end.p0(i64 500, ptr nonnull %data)
27  br label %cleanup1
28
29cleanup1:
30  call void @llvm.lifetime.end.p0(i64 500, ptr nonnull %data)
31  br label %cleanup
32
33else:
34  call void @llvm.lifetime.start.p0(i64 500, ptr nonnull %data2)
35  call void @consume(ptr %data2)
36  %suspend.value2 = call i8 @llvm.coro.suspend(token none, i1 false)
37  switch i8 %suspend.value2, label %coro.ret [i8 0, label %resume2
38                                              i8 1, label %cleanup2]
39
40resume2:
41  call void @llvm.lifetime.end.p0(i64 500, ptr nonnull %data2)
42  br label %cleanup2
43
44cleanup2:
45  call void @llvm.lifetime.end.p0(i64 500, ptr nonnull %data2)
46  br label %cleanup
47
48cleanup:
49  %mem = call ptr @llvm.coro.free(token %id, ptr %hdl)
50  call void @free(ptr %mem)
51  br label %coro.ret
52coro.ret:
53  call i1 @llvm.coro.end(ptr %hdl, i1 0, token none)
54  ret ptr %hdl
55}
56
57; CHECK-LABEL: @f(
58; CHECK: call ptr @malloc(i32 520)
59
60declare ptr @llvm.coro.free(token, ptr)
61declare i32 @llvm.coro.size.i32()
62declare i8  @llvm.coro.suspend(token, i1)
63declare void @llvm.coro.resume(ptr)
64declare void @llvm.coro.destroy(ptr)
65
66declare token @llvm.coro.id(i32, ptr, ptr, ptr)
67declare i1 @llvm.coro.alloc(token)
68declare ptr @llvm.coro.begin(token, ptr)
69declare i1 @llvm.coro.end(ptr, i1, token)
70
71declare noalias ptr @malloc(i32)
72declare double @print(double)
73declare void @free(ptr)
74
75declare void @llvm.lifetime.start.p0(i64, ptr nocapture)
76declare void @llvm.lifetime.end.p0(i64, ptr nocapture)
77