llvm/test/Transforms/Coroutines/coro-split-musttail6.ll

   1 ; Tests that sinked lifetime markers wouldn't provent optimization
   2 ; to convert a resuming call to a musttail call.
   3 ; The difference between this and coro-split-musttail5.ll is that there is
   4 ; an extra bitcast instruction in the path, which makes it harder to
   5 ; optimize.
   6 ; RUN: opt < %s -passes='cgscc(coro-split),simplifycfg,early-cse' -S | FileCheck %s
   7 ; RUN: opt < %s -passes='pgo-instr-gen,cgscc(coro-split),simplifycfg,early-cse' -S | FileCheck %s
   8
   9 declare void @fakeresume1(ptr align 8)
  10
  11 define void @g() #0 {
  12 entry:
  13   %id = call token @llvm.coro.id(i32 0, ptr null, ptr null, ptr null)
  14   %alloc = call ptr @malloc(i64 16) #3
  15   %alloc.var = alloca i64
  16   call void @llvm.lifetime.start.p0(i64 1, ptr %alloc.var)
  17   %vFrame = call noalias nonnull ptr @llvm.coro.begin(token %id, ptr %alloc)
  18
  19   %save = call token @llvm.coro.save(ptr null)
  20   %suspend = call i8 @llvm.coro.suspend(token %save, i1 false)
  21
  22   switch i8 %suspend, label %exit [
  23     i8 0, label %await.suspend
  24     i8 1, label %exit
  25   ]
  26 await.suspend:
  27   %save2 = call token @llvm.coro.save(ptr null)
  28   call fastcc void @fakeresume1(ptr align 8 null)
  29   %suspend2 = call i8 @llvm.coro.suspend(token %save2, i1 false)
  30   switch i8 %suspend2, label %exit [
  31     i8 0, label %await.ready
  32     i8 1, label %exit
  33   ]
  34 await.ready:
  35   call void @consume(ptr %alloc.var)
  36   call void @llvm.lifetime.end.p0(i64 1, ptr %alloc.var)
  37   br label %exit
  38 exit:
  39   call i1 @llvm.coro.end(ptr null, i1 false, token none)
  40   ret void
  41 }
  42
  43 ; Verify that in the resume part resume call is marked with musttail.
  44 ; CHECK-LABEL: @g.resume(
  45 ; CHECK:      musttail call fastcc void @fakeresume1(ptr align 8 null)
  46 ; CHECK-NEXT: ret void
  47
  48 ; It has a cleanup bb.
  49 define void @f() #0 {
  50 entry:
  51   %id = call token @llvm.coro.id(i32 0, ptr null, ptr null, ptr null)
  52   %alloc = call ptr @malloc(i64 16) #3
  53   %alloc.var = alloca i64
  54   call void @llvm.lifetime.start.p0(i64 1, ptr %alloc.var)
  55   %vFrame = call noalias nonnull ptr @llvm.coro.begin(token %id, ptr %alloc)
  56
  57   %save = call token @llvm.coro.save(ptr null)
  58   %suspend = call i8 @llvm.coro.suspend(token %save, i1 false)
  59
  60   switch i8 %suspend, label %exit [
  61     i8 0, label %await.suspend
  62     i8 1, label %exit
  63   ]
  64 await.suspend:
  65   %save2 = call token @llvm.coro.save(ptr null)
  66   call fastcc void @fakeresume1(ptr align 8 null)
  67   %suspend2 = call i8 @llvm.coro.suspend(token %save2, i1 false)
  68   switch i8 %suspend2, label %exit [
  69     i8 0, label %await.ready
  70     i8 1, label %cleanup
  71   ]
  72 await.ready:
  73   call void @consume(ptr %alloc.var)
  74   call void @llvm.lifetime.end.p0(i64 1, ptr %alloc.var)
  75   br label %exit
  76
  77 cleanup:
  78   %free.handle = call ptr @llvm.coro.free(token %id, ptr %vFrame)
  79   %.not = icmp eq ptr %free.handle, null
  80   br i1 %.not, label %exit, label %coro.free
  81
  82 coro.free:
  83   call void @delete(ptr nonnull %free.handle) #2
  84   br label %exit
  85
  86 exit:
  87   call i1 @llvm.coro.end(ptr null, i1 false, token none)
  88   ret void
  89 }
  90
  91 ; FIXME: The fakeresume1 here should be marked as musttail.
  92 ; Verify that in the resume part resume call is marked with musttail.
  93 ; CHECK-LABEL: @f.resume(
  94 ; CHECK:      musttail call fastcc void @fakeresume1(ptr align 8 null)
  95 ; CHECK-NEXT: ret void
  96
  97 declare token @llvm.coro.id(i32, ptr readnone, ptr nocapture readonly, ptr) #1
  98 declare i1 @llvm.coro.alloc(token) #2
  99 declare i64 @llvm.coro.size.i64() #3
 100 declare ptr @llvm.coro.begin(token, ptr writeonly) #2
 101 declare token @llvm.coro.save(ptr) #2
 102 declare ptr @llvm.coro.frame() #3
 103 declare i8 @llvm.coro.suspend(token, i1) #2
 104 declare ptr @llvm.coro.free(token, ptr nocapture readonly) #1
 105 declare i1 @llvm.coro.end(ptr, i1, token) #2
 106 declare ptr @llvm.coro.subfn.addr(ptr nocapture readonly, i8) #1
 107 declare ptr @malloc(i64)
 108 declare void @delete(ptr nonnull) #2
 109 declare void @consume(ptr)
 110 declare void @llvm.lifetime.start.p0(i64, ptr nocapture)
 111 declare void @llvm.lifetime.end.p0(i64, ptr nocapture)
 112
 113 attributes #0 = { presplitcoroutine }
 114 attributes #1 = { argmemonly nounwind readonly }
 115 attributes #2 = { nounwind }
 116 attributes #3 = { nounwind readnone }