llvm/test/Transforms/Coroutines/coro-frame-reuse-alloca-02.ll

   1 ; Tests that variables of different type in a Corotuine whose lifetime range is not overlapping each other
   2 ; re-use the same slot in Coroutine frame.
   3 ; RUN: opt < %s -passes='cgscc(coro-split<reuse-storage>),simplifycfg,early-cse' -S | FileCheck %s
   4 %"struct.task::promise_type" = type { i8 }
   5 %struct.awaitable = type { i8 }
   6 %struct.big_structure = type { [500 x i8] }
   7 %struct.big_structure.2 = type { [300 x i8] }
   8 declare ptr @malloc(i64)
   9 declare void @consume(ptr)
  10 declare void @consume.2(ptr)
  11 define void @a(i1 zeroext %cond) presplitcoroutine {
  12 entry:
  13   %__promise = alloca %"struct.task::promise_type", align 1
  14   %a = alloca %struct.big_structure, align 1
  15   %ref.tmp7 = alloca %struct.awaitable, align 1
  16   %b = alloca %struct.big_structure.2, align 1
  17   %ref.tmp18 = alloca %struct.awaitable, align 1
  18   %0 = call token @llvm.coro.id(i32 16, ptr nonnull %__promise, ptr @a, ptr null)
  19   br label %init.ready
  20 init.ready:
  21   %1 = call noalias nonnull ptr @llvm.coro.begin(token %0, ptr null)
  22   call void @llvm.lifetime.start.p0(i64 1, ptr nonnull %__promise)
  23   br i1 %cond, label %if.then, label %if.else
  24 if.then:
  25   call void @llvm.lifetime.start.p0(i64 500, ptr nonnull %a)
  26   call void @consume(ptr nonnull %a)
  27   %save = call token @llvm.coro.save(ptr null)
  28   %suspend = call i8 @llvm.coro.suspend(token %save, i1 false)
  29   switch i8 %suspend, label %coro.ret [
  30     i8 0, label %await.ready
  31     i8 1, label %cleanup1
  32   ]
  33 await.ready:
  34   call void @llvm.lifetime.end.p0(i64 500, ptr nonnull %a)
  35   br label %cleanup1
  36 if.else:
  37   call void @llvm.lifetime.start.p0(i64 300, ptr nonnull %b)
  38   call void @consume.2(ptr nonnull %b)
  39   %save2 = call token @llvm.coro.save(ptr null)
  40   %suspend2 = call i8 @llvm.coro.suspend(token %save2, i1 false)
  41   switch i8 %suspend2, label %coro.ret [
  42     i8 0, label %await2.ready
  43     i8 1, label %cleanup2
  44   ]
  45 await2.ready:
  46   call void @llvm.lifetime.end.p0(i64 300, ptr nonnull %b)
  47   br label %cleanup2
  48 cleanup1:
  49   call void @llvm.lifetime.end.p0(i64 500, ptr nonnull %a)
  50   br label %cleanup
  51 cleanup2:
  52   call void @llvm.lifetime.end.p0(i64 300, ptr nonnull %b)
  53   br label %cleanup
  54 cleanup:
  55   call ptr @llvm.coro.free(token %0, ptr %1)
  56   br label %coro.ret
  57 coro.ret:
  58   call i1 @llvm.coro.end(ptr null, i1 false, token none)
  59   ret void
  60 }
  61 ; CHECK:       %a.Frame = type { ptr, ptr, %"struct.task::promise_type", %struct.big_structure, i1 }
  62 ; CHECK-LABEL: @a.resume(
  63
  64 declare token @llvm.coro.id(i32, ptr readnone, ptr nocapture readonly, ptr)
  65 declare i1 @llvm.coro.alloc(token) #3
  66 declare i64 @llvm.coro.size.i64() #5
  67 declare ptr @llvm.coro.begin(token, ptr writeonly) #3
  68 declare token @llvm.coro.save(ptr) #3
  69 declare ptr @llvm.coro.frame() #5
  70 declare i8 @llvm.coro.suspend(token, i1) #3
  71 declare ptr @llvm.coro.free(token, ptr nocapture readonly) #2
  72 declare i1 @llvm.coro.end(ptr, i1, token) #3
  73 declare void @llvm.lifetime.start.p0(i64, ptr nocapture) #4
  74 declare void @llvm.lifetime.end.p0(i64, ptr nocapture) #4