xref: /llvm-project/llvm/test/Transforms/Coroutines/coro-retcon-remat.ll (revision 51d5d7bbae92493a5bfa7cc6b519de8a5bb32fdb) 
1; Check that a remat that inserts rematerialized instructions in the single predecessor block works
2; as expected
3; RUN: opt < %s -O0 -S | FileCheck %s
4
5; CHECK: %f.Frame = type { i32 }
6
7define { ptr, i32 } @f(ptr %buffer, i32 %n) {
8entry:
9  %id = call token @llvm.coro.id.retcon(i32 8, i32 4, ptr %buffer, ptr @f_prototype, ptr @allocate, ptr @deallocate)
10  %hdl = call ptr @llvm.coro.begin(token %id, ptr null)
11  br label %loop
12
13loop:
14  %n.val = phi i32 [ %n, %entry ], [ %inc, %resume1 ]
15  call void @print(i32 %n.val)
16  %inc1 = add i32 %n.val, 1
17  %inc2 = add i32 %inc1, 2
18  %inc3 = add i32 %inc2, 3
19  %inc4 = add i32 %inc3, 4
20  %inc5 = add i32 %inc4, 5
21  %inc6 = add i32 %inc5, 6
22  %unwind0 = call i1 (...) @llvm.coro.suspend.retcon.i1(i32 %inc6)
23  br i1 %unwind0, label %cleanup, label %resume
24
25resume:
26  %unwind1 = call i1 (...) @llvm.coro.suspend.retcon.i1(i32 %inc6)
27  br i1 %unwind1, label %cleanup, label %resume1
28
29resume1:
30  %inc = add i32 %n.val, 1
31  br label %loop
32
33cleanup:
34  call i1 @llvm.coro.end(ptr %hdl, i1 0, token none)
35  unreachable
36}
37
38declare token @llvm.coro.id.retcon(i32, i32, ptr, ptr, ptr, ptr)
39declare ptr @llvm.coro.begin(token, ptr)
40declare i1 @llvm.coro.suspend.retcon.i1(...)
41declare i1 @llvm.coro.end(ptr, i1, token)
42declare ptr @llvm.coro.prepare.retcon(ptr)
43
44declare { ptr, i32 } @f_prototype(ptr, i1 zeroext)
45
46declare noalias ptr @allocate(i32 %size)
47declare void @deallocate(ptr %ptr)
48
49declare void @print(i32)
50