xref: /llvm-project/llvm/test/Transforms/Coroutines/coro-zero-alloca.ll (revision 51d5d7bbae92493a5bfa7cc6b519de8a5bb32fdb) 
1; RUN: opt < %s -passes='cgscc(coro-split),simplifycfg,early-cse' -S | FileCheck %s
2
3declare ptr @malloc(i64)
4declare void @free(ptr)
5declare void @usePointer(ptr)
6declare void @usePointer2(ptr)
7
8declare token @llvm.coro.id(i32, ptr readnone, ptr nocapture readonly, ptr)
9declare i64 @llvm.coro.size.i64()
10declare ptr @llvm.coro.begin(token, ptr writeonly)
11declare i8 @llvm.coro.suspend(token, i1)
12declare i1 @llvm.coro.end(ptr, i1, token)
13declare ptr @llvm.coro.free(token, ptr nocapture readonly)
14declare token @llvm.coro.save(ptr)
15
16define void @foo() presplitcoroutine {
17entry:
18  %a0 = alloca [0 x i8]
19  %a1 = alloca i32
20  %a2 = alloca [0 x i8]
21  %a3 = alloca [0 x i8]
22  %a4 = alloca i16
23  %a5 = alloca [0 x i8]
24  %coro.id = call token @llvm.coro.id(i32 0, ptr null, ptr null, ptr null)
25  %coro.size = call i64 @llvm.coro.size.i64()
26  %coro.alloc = call ptr @malloc(i64 %coro.size)
27  %coro.state = call ptr @llvm.coro.begin(token %coro.id, ptr %coro.alloc)
28  %coro.save = call token @llvm.coro.save(ptr %coro.state)
29  %call.suspend = call i8 @llvm.coro.suspend(token %coro.save, i1 false)
30  switch i8 %call.suspend, label %suspend [
31    i8 0, label %wakeup
32    i8 1, label %cleanup
33  ]
34
35wakeup:                                           ; preds = %entry
36  call void @usePointer(ptr %a0)
37  call void @usePointer(ptr %a1)
38  call void @usePointer(ptr %a2)
39  call void @usePointer(ptr %a3)
40  call void @usePointer(ptr %a4)
41  call void @usePointer2(ptr %a5)
42  br label %cleanup
43
44suspend:                                          ; preds = %cleanup, %entry
45  %unused = call i1 @llvm.coro.end(ptr %coro.state, i1 false, token none)
46  ret void
47
48cleanup:                                          ; preds = %wakeup, %entry
49  %coro.memFree = call ptr @llvm.coro.free(token %coro.id, ptr %coro.state)
50  call void @free(ptr %coro.memFree)
51  br label %suspend
52}
53
54; CHECK:       %foo.Frame = type { ptr, ptr, i32, i16, i1 }
55
56; CHECK-LABEL: @foo.resume(
57; CHECK-NEXT:  entry.resume:
58; CHECK-NEXT:    [[A1_RELOAD_ADDR:%.*]] = getelementptr inbounds [[FOO_FRAME:%foo.Frame]], ptr [[FRAMEPTR:%.*]], i32 0, i32 2
59; CHECK-NEXT:    [[A4_RELOAD_ADDR:%.*]] = getelementptr inbounds [[FOO_FRAME]], ptr [[FRAMEPTR]], i32 0, i32 3
60; CHECK-NEXT:    call void @usePointer(ptr [[FRAMEPTR]])
61; CHECK-NEXT:    call void @usePointer(ptr [[A1_RELOAD_ADDR]])
62; CHECK-NEXT:    call void @usePointer(ptr [[FRAMEPTR]])
63; CHECK-NEXT:    call void @usePointer(ptr [[FRAMEPTR]])
64; CHECK-NEXT:    call void @usePointer(ptr [[A4_RELOAD_ADDR]])
65; CHECK-NEXT:    call void @usePointer2(ptr [[FRAMEPTR]])
66; CHECK-NEXT:    call void @free(ptr [[FRAMEPTR]])
67; CHECK-NEXT:    ret void
68