llvm/test/Transforms/Coroutines/coro-eh-aware-edge-split-00.ll

   1 ; Check that we can handle edge splits leading into a landingpad
   2 ; RUN: opt < %s -passes='cgscc(coro-split),simplifycfg,early-cse' -S | FileCheck %s
   3
   4 target datalayout = "e-m:e-i64:64-f80:128-n8:16:32:64-S128"
   5 target triple = "x86_64-unknown-linux-gnu"
   6
   7 ; CHECK-LABEL: define internal fastcc void @f.resume(
   8 define void @f(i1 %cond) presplitcoroutine personality i32 0 {
   9 entry:
  10   %id = call token @llvm.coro.id(i32 16, ptr null, ptr null, ptr null)
  11   %size = tail call i64 @llvm.coro.size.i64()
  12   %alloc = call ptr @malloc(i64 %size)
  13   %hdl = call ptr @llvm.coro.begin(token %id, ptr %alloc)
  14   %sp = call i8 @llvm.coro.suspend(token none, i1 false)
  15   switch i8 %sp, label %coro.ret [
  16     i8 0, label %resume
  17     i8 1, label %cleanup
  18   ]
  19
  20 resume:
  21   br i1 %cond, label %invoke1, label %invoke2
  22
  23 invoke1:
  24   invoke void @may_throw1()
  25           to label %unreach unwind label %pad.with.phi
  26 invoke2:
  27   invoke void @may_throw2()
  28           to label %unreach unwind label %pad.with.phi
  29
  30 ; Verify that we cloned landing pad on every edge and inserted a reload of the spilled value
  31
  32 ; CHECK: pad.with.phi.from.invoke2:
  33 ; CHECK:   %0 = landingpad { ptr, i32 }
  34 ; CHECK:           catch ptr null
  35 ; CHECK:   br label %pad.with.phi
  36
  37 ; CHECK: pad.with.phi.from.invoke1:
  38 ; CHECK:   %1 = landingpad { ptr, i32 }
  39 ; CHECK:           catch ptr null
  40 ; CHECK:   br label %pad.with.phi
  41
  42 ; CHECK: pad.with.phi:
  43 ; CHECK:   %val = phi i32 [ 0, %pad.with.phi.from.invoke1 ], [ 1, %pad.with.phi.from.invoke2 ]
  44 ; CHECK:   %lp = phi { ptr, i32 } [ %0, %pad.with.phi.from.invoke2 ], [ %1, %pad.with.phi.from.invoke1 ]
  45 ; CHECK:   %exn = extractvalue { ptr, i32 } %lp, 0
  46 ; CHECK:   call ptr @__cxa_begin_catch(ptr %exn)
  47 ; CHECK:   call void @use_val(i32 %val)
  48 ; CHECK:   call void @__cxa_end_catch()
  49 ; CHECK:   call void @free(ptr %hdl)
  50 ; CHECK:   ret void
  51
  52 pad.with.phi:
  53   %val = phi i32 [ 0, %invoke1 ], [ 1, %invoke2 ]
  54   %lp = landingpad { ptr, i32 }
  55           catch ptr null
  56   %exn = extractvalue { ptr, i32 } %lp, 0
  57   call ptr @__cxa_begin_catch(ptr %exn)
  58   call void @use_val(i32 %val)
  59   call void @__cxa_end_catch()
  60   br label %cleanup
  61
  62 cleanup:                                        ; preds = %invoke.cont15, %if.else, %if.then, %ehcleanup21, %init.suspend
  63   %mem = call ptr @llvm.coro.free(token %id, ptr %hdl)
  64   call void @free(ptr %mem)
  65   br label %coro.ret
  66
  67 coro.ret:
  68   call i1 @llvm.coro.end(ptr null, i1 false, token none)
  69   ret void
  70
  71 unreach:
  72   unreachable
  73 }
  74
  75 ; Function Attrs: argmemonly nounwind readonly
  76 declare token @llvm.coro.id(i32, ptr readnone, ptr nocapture readonly, ptr)
  77 declare noalias ptr @malloc(i64)
  78 declare i64 @llvm.coro.size.i64()
  79 declare ptr @llvm.coro.begin(token, ptr writeonly)
  80
  81 ; Function Attrs: nounwind
  82 declare token @llvm.coro.save(ptr)
  83 declare i8 @llvm.coro.suspend(token, i1)
  84
  85 ; Function Attrs: argmemonly nounwind
  86 declare void @may_throw1()
  87 declare void @may_throw2()
  88
  89 declare ptr @__cxa_begin_catch(ptr)
  90
  91 declare void @use_val(i32)
  92 declare void @__cxa_end_catch()
  93
  94 ; Function Attrs: nounwind
  95 declare i1 @llvm.coro.end(ptr, i1, token)
  96 declare void @free(ptr)
  97 declare ptr @llvm.coro.free(token, ptr nocapture readonly)