llvm/test/Transforms/Coroutines/coro-retcon-remat.ll

   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
   7 define { ptr, i32 } @f(ptr %buffer, i32 %n) {
   8 entry:
   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
  13 loop:
  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
  25 resume:
  26   %unwind1 = call i1 (...) @llvm.coro.suspend.retcon.i1(i32 %inc6)
  27   br i1 %unwind1, label %cleanup, label %resume1
  28
  29 resume1:
  30   %inc = add i32 %n.val, 1
  31   br label %loop
  32
  33 cleanup:
  34   call i1 @llvm.coro.end(ptr %hdl, i1 0, token none)
  35   unreachable
  36 }
  37
  38 declare token @llvm.coro.id.retcon(i32, i32, ptr, ptr, ptr, ptr)
  39 declare ptr @llvm.coro.begin(token, ptr)
  40 declare i1 @llvm.coro.suspend.retcon.i1(...)
  41 declare i1 @llvm.coro.end(ptr, i1, token)
  42 declare ptr @llvm.coro.prepare.retcon(ptr)
  43
  44 declare { ptr, i32 } @f_prototype(ptr, i1 zeroext)
  45
  46 declare noalias ptr @allocate(i32 %size)
  47 declare void @deallocate(ptr %ptr)
  48
  49 declare void @print(i32)