llvm/test/Transforms/Coroutines/coro-split-final-suspend.ll

   1 ; Tests that we'll generate the store to the final suspend index if we see the unwind coro end.
   2 ; RUN: opt < %s -passes='cgscc(coro-split),simplifycfg,early-cse' -S | FileCheck %s
   3
   4 define ptr @unwind_coro_end() presplitcoroutine personality i32 3 {
   5 entry:
   6   %id = call token @llvm.coro.id(i32 0, ptr null, ptr null, ptr null)
   7   %hdl = call ptr @llvm.coro.begin(token %id, ptr null)
   8   call void @print(i32 0)
   9   br label %init
  10
  11 init:
  12   %initial_suspend = call i8 @llvm.coro.suspend(token none, i1 false)
  13   switch i8 %initial_suspend, label %init_suspend [i8 0, label %init_resume
  14                                                    i8 1, label %init_suspend]
  15
  16 init_suspend:
  17   ret ptr %hdl
  18
  19 init_resume:
  20   invoke void @print(i32 1)
  21           to label %final_suspend unwind label %lpad
  22
  23 final_suspend:
  24   %0 = call i8 @llvm.coro.suspend(token none, i1 true)
  25   switch i8 %0, label %suspend [i8 0, label %resume
  26                                 i8 1, label %suspend]
  27 resume:
  28   invoke void @print(i32 1) to label %suspend unwind label %lpad
  29
  30 suspend:
  31   call i1 @llvm.coro.end(ptr %hdl, i1 false, token none)
  32   call void @print(i32 0)
  33   ret ptr %hdl
  34
  35 lpad:
  36   %lpval = landingpad { ptr, i32 }
  37      cleanup
  38
  39   call void @print(i32 2)
  40   %need.resume = call i1 @llvm.coro.end(ptr null, i1 true, token none)
  41   br i1 %need.resume, label %eh.resume, label %cleanup.cont
  42
  43 cleanup.cont:
  44   call void @print(i32 3)
  45   br label %eh.resume
  46
  47 eh.resume:
  48   resume { ptr, i32 } %lpval
  49 }
  50
  51 ; Tests that we need to store the final index if we see unwind coro end.
  52 ; CHECK: define{{.*}}@unwind_coro_end.resume
  53 ; CHECK: invoke{{.*}}print
  54 ; CHECK-NEXT: to label %[[RESUME:.*]] unwind label %[[LPAD:.*]]
  55
  56 ; CHECK: [[RESUME]]:
  57 ; CHECK-NOT: {{.*:}}
  58 ; CHECK: store ptr null, ptr %hdl
  59 ; CHECK-NOT: {{.*:}}
  60 ; CHECK: store i1 true, ptr %index.addr
  61
  62 ; CHECK: [[LPAD]]:
  63 ; CHECK-NOT: {{.*:}}
  64 ; CHECK: store ptr null, ptr %hdl
  65 ; CHECK-NOT: {{.*:}}
  66 ; CHECK: store i1 true, ptr %index.addr
  67
  68 ; Tests the use of final index in the destroy function.
  69 ; CHECK: define{{.*}}@unwind_coro_end.destroy
  70 ; CHECK: %[[INDEX:.+]] = load i1, ptr %index.addr
  71 ; CHECK-NEXT: switch i1 %[[INDEX]],
  72
  73 define ptr @nounwind_coro_end(i1 %val) presplitcoroutine personality i32 3 {
  74 entry:
  75   %id = call token @llvm.coro.id(i32 0, ptr null, ptr null, ptr null)
  76   %hdl = call ptr @llvm.coro.begin(token %id, ptr null)
  77   call void @print(i32 0)
  78   br label %init
  79
  80 init:
  81   %initial_suspend = call i8 @llvm.coro.suspend(token none, i1 false)
  82   switch i8 %initial_suspend, label %init_suspend [i8 0, label %init_resume
  83                                                    i8 1, label %init_suspend]
  84
  85 init_suspend:
  86   ret ptr %hdl
  87
  88 init_resume:
  89   br label %susp
  90
  91 susp:
  92   %0 = call i8 @llvm.coro.suspend(token none, i1 true)
  93   switch i8 %0, label %suspend [i8 0, label %resume
  94                                 i8 1, label %suspend]
  95 resume:
  96   call void @print(i32 1)
  97   br label %suspend
  98
  99 suspend:
 100   call i1 @llvm.coro.end(ptr %hdl, i1 false, token none)
 101   call void @print(i32 0)
 102   ret ptr %hdl
 103 }
 104
 105 ; Tests that we can omit to store the final suspend index if we don't
 106 ; see unwind coro end.
 107 ; CHECK: define{{.*}}@nounwind_coro_end.resume
 108 ; CHECK-NOT: store i1 true, ptr %index.addr
 109 ; CHECK: }
 110
 111 ; Tests that we judge the final suspend case by the nullness of resume function.
 112 ; CHECK: define{{.*}}@nounwind_coro_end.destroy
 113 ; CHECK:  %[[RESUME_FN:.+]] = load ptr, ptr %hdl, align 8
 114 ; CHECK:  %{{.*}} = icmp eq ptr %[[RESUME_FN]], null
 115
 116 declare ptr @llvm.coro.free(token, ptr)
 117 declare i32 @llvm.coro.size.i32()
 118 declare i8  @llvm.coro.suspend(token, i1)
 119 declare void @llvm.coro.resume(ptr)
 120 declare void @llvm.coro.destroy(ptr)
 121
 122 declare token @llvm.coro.id(i32, ptr, ptr, ptr)
 123 declare ptr @llvm.coro.alloc(token)
 124 declare ptr @llvm.coro.begin(token, ptr)
 125 declare i1 @llvm.coro.end(ptr, i1, token)
 126
 127 declare noalias ptr @malloc(i32)
 128 declare void @print(i32)
 129 declare void @free(ptr)