xref: /llvm-project/llvm/test/Transforms/Coroutines/coro-retcon-resume-values2.ll (revision 51d5d7bbae92493a5bfa7cc6b519de8a5bb32fdb) 
1; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --include-generated-funcs
2; RUN: opt < %s -passes='cgscc(coro-split),simplifycfg,early-cse,simplifycfg,coro-cleanup' -S | FileCheck %s
3
4define ptr @f(ptr %buffer, i32 %n) presplitcoroutine {
5entry:
6  %id = call token @llvm.coro.id.retcon(i32 8, i32 4, ptr %buffer, ptr @prototype, ptr @allocate, ptr @deallocate)
7  %hdl = call ptr @llvm.coro.begin(token %id, ptr null)
8  %value0 = call i32 (...) @llvm.coro.suspend.retcon.i32()
9  %sum0 = call i32 @add(i32 %n, i32 %value0)
10  %value1 = call i32 (...) @llvm.coro.suspend.retcon.i32()
11  %sum1 = call i32 @add(i32 %sum0, i32 %value0)
12  %sum2 = call i32 @add(i32 %sum1, i32 %value1)
13  %value2 = call i32 (...) @llvm.coro.suspend.retcon.i32()
14  %sum3 = call i32 @add(i32 %sum2, i32 %value0)
15  %sum4 = call i32 @add(i32 %sum3, i32 %value1)
16  %sum5 = call i32 @add(i32 %sum4, i32 %value2)
17  call void @print(i32 %sum5)
18  call i1 @llvm.coro.end(ptr %hdl, i1 0, token none)
19  unreachable
20}
21
22declare token @llvm.coro.id.retcon(i32, i32, ptr, ptr, ptr, ptr)
23declare ptr @llvm.coro.begin(token, ptr)
24declare i32 @llvm.coro.suspend.retcon.i32(...)
25declare i1 @llvm.coro.end(ptr, i1, token)
26declare ptr @llvm.coro.prepare.retcon(ptr)
27
28declare ptr @prototype(ptr, i32)
29
30declare noalias ptr @allocate(i32 %size)
31declare void @deallocate(ptr %ptr)
32
33declare i32 @add(i32, i32)
34declare void @print(i32)
35
36; CHECK-LABEL: @f(
37; CHECK-NEXT:  entry:
38; CHECK-NEXT:    [[TMP0:%.*]] = call ptr @allocate(i32 20)
39; CHECK-NEXT:    store ptr [[TMP0]], ptr [[BUFFER:%.*]], align 8
40; CHECK-NEXT:    store i32 [[N:%.*]], ptr [[TMP0]], align 4
41; CHECK-NEXT:    ret ptr @f.resume.0
42;
43;
44; CHECK-LABEL: @f.resume.0(
45; CHECK-NEXT:  entryresume.0:
46; CHECK-NEXT:    [[TMP2:%.*]] = load ptr, ptr [[TMP0:%.*]], align 8
47; CHECK-NEXT:    [[VALUE0_SPILL_ADDR:%.*]] = getelementptr inbounds [[F_FRAME:%.*]], ptr [[TMP2]], i32 0, i32 1
48; CHECK-NEXT:    store i32 [[TMP1:%.*]], ptr [[VALUE0_SPILL_ADDR]], align 4
49; CHECK-NEXT:    [[N_RELOAD:%.*]] = load i32, ptr [[TMP2]], align 4
50; CHECK-NEXT:    [[SUM0:%.*]] = call i32 @add(i32 [[N_RELOAD]], i32 [[TMP1]])
51; CHECK-NEXT:    [[SUM0_SPILL_ADDR:%.*]] = getelementptr inbounds [[F_FRAME]], ptr [[TMP2]], i32 0, i32 2
52; CHECK-NEXT:    store i32 [[SUM0]], ptr [[SUM0_SPILL_ADDR]], align 4
53; CHECK-NEXT:    ret ptr @f.resume.1
54;
55;
56; CHECK-LABEL: @f.resume.1(
57; CHECK-NEXT:  entryresume.1:
58; CHECK-NEXT:    [[TMP2:%.*]] = load ptr, ptr [[TMP0:%.*]], align 8
59; CHECK-NEXT:    [[VALUE1_SPILL_ADDR:%.*]] = getelementptr inbounds [[F_FRAME:%.*]], ptr [[TMP2]], i32 0, i32 3
60; CHECK-NEXT:    store i32 [[TMP1:%.*]], ptr [[VALUE1_SPILL_ADDR]], align 4
61; CHECK-NEXT:    [[SUM0_RELOAD_ADDR:%.*]] = getelementptr inbounds [[F_FRAME]], ptr [[TMP2]], i32 0, i32 2
62; CHECK-NEXT:    [[SUM0_RELOAD:%.*]] = load i32, ptr [[SUM0_RELOAD_ADDR]], align 4
63; CHECK-NEXT:    [[VALUE0_RELOAD_ADDR5:%.*]] = getelementptr inbounds [[F_FRAME]], ptr [[TMP2]], i32 0, i32 1
64; CHECK-NEXT:    [[VALUE0_RELOAD6:%.*]] = load i32, ptr [[VALUE0_RELOAD_ADDR5]], align 4
65; CHECK-NEXT:    [[SUM1:%.*]] = call i32 @add(i32 [[SUM0_RELOAD]], i32 [[VALUE0_RELOAD6]])
66; CHECK-NEXT:    [[SUM2:%.*]] = call i32 @add(i32 [[SUM1]], i32 [[TMP1]])
67; CHECK-NEXT:    [[SUM2_SPILL_ADDR:%.*]] = getelementptr inbounds [[F_FRAME]], ptr [[TMP2]], i32 0, i32 4
68; CHECK-NEXT:    store i32 [[SUM2]], ptr [[SUM2_SPILL_ADDR]], align 4
69; CHECK-NEXT:    ret ptr @f.resume.2
70;
71;
72; CHECK-LABEL: @f.resume.2(
73; CHECK-NEXT:  entryresume.2:
74; CHECK-NEXT:    [[TMP2:%.*]] = load ptr, ptr [[TMP0:%.*]], align 8
75; CHECK-NEXT:    [[SUM2_RELOAD_ADDR:%.*]] = getelementptr inbounds [[F_FRAME:%.*]], ptr [[TMP2]], i32 0, i32 4
76; CHECK-NEXT:    [[SUM2_RELOAD:%.*]] = load i32, ptr [[SUM2_RELOAD_ADDR]], align 4
77; CHECK-NEXT:    [[VALUE1_RELOAD_ADDR:%.*]] = getelementptr inbounds [[F_FRAME]], ptr [[TMP2]], i32 0, i32 3
78; CHECK-NEXT:    [[VALUE1_RELOAD:%.*]] = load i32, ptr [[VALUE1_RELOAD_ADDR]], align 4
79; CHECK-NEXT:    [[VALUE0_RELOAD_ADDR:%.*]] = getelementptr inbounds [[F_FRAME]], ptr [[TMP2]], i32 0, i32 1
80; CHECK-NEXT:    [[VALUE0_RELOAD:%.*]] = load i32, ptr [[VALUE0_RELOAD_ADDR]], align 4
81; CHECK-NEXT:    [[SUM3:%.*]] = call i32 @add(i32 [[SUM2_RELOAD]], i32 [[VALUE0_RELOAD]])
82; CHECK-NEXT:    [[SUM4:%.*]] = call i32 @add(i32 [[SUM3]], i32 [[VALUE1_RELOAD]])
83; CHECK-NEXT:    [[SUM5:%.*]] = call i32 @add(i32 [[SUM4]], i32 [[TMP1:%.*]])
84; CHECK-NEXT:    call void @print(i32 [[SUM5]])
85; CHECK-NEXT:    call void @deallocate(ptr [[TMP2]])
86; CHECK-NEXT:    ret ptr null
87;
88