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 i8* @malloc(i64)
   9 declare void @consume(%struct.big_structure*)
  10 declare void @consume.2(%struct.big_structure.2*)
  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 = getelementptr inbounds %"struct.task::promise_type", %"struct.task::promise_type"* %__promise, i64 0, i32 0
  19   %1 = call token @llvm.coro.id(i32 16, i8* nonnull %0, i8* bitcast (void (i1)* @a to i8*), i8* null)
  20   br label %init.ready
  21 init.ready:
  22   %2 = call noalias nonnull i8* @llvm.coro.begin(token %1, i8* null)
  23   call void @llvm.lifetime.start.p0i8(i64 1, i8* nonnull %0)
  24   br i1 %cond, label %if.then, label %if.else
  25 if.then:
  26   %3 = getelementptr inbounds %struct.big_structure, %struct.big_structure* %a, i64 0, i32 0, i64 0
  27   call void @llvm.lifetime.start.p0i8(i64 500, i8* nonnull %3)
  28   call void @consume(%struct.big_structure* nonnull %a)
  29   %save = call token @llvm.coro.save(i8* null)
  30   %suspend = call i8 @llvm.coro.suspend(token %save, i1 false)
  31   switch i8 %suspend, label %coro.ret [
  32     i8 0, label %await.ready
  33     i8 1, label %cleanup1
  34   ]
  35 await.ready:
  36   call void @llvm.lifetime.end.p0i8(i64 500, i8* nonnull %3)
  37   br label %cleanup1
  38 if.else:
  39   %4 = getelementptr inbounds %struct.big_structure.2, %struct.big_structure.2* %b, i64 0, i32 0, i64 0
  40   call void @llvm.lifetime.start.p0i8(i64 300, i8* nonnull %4)
  41   call void @consume.2(%struct.big_structure.2* nonnull %b)
  42   %save2 = call token @llvm.coro.save(i8* null)
  43   %suspend2 = call i8 @llvm.coro.suspend(token %save2, i1 false)
  44   switch i8 %suspend2, label %coro.ret [
  45     i8 0, label %await2.ready
  46     i8 1, label %cleanup2
  47   ]
  48 await2.ready:
  49   call void @llvm.lifetime.end.p0i8(i64 300, i8* nonnull %4)
  50   br label %cleanup2
  51 cleanup1:
  52   call void @llvm.lifetime.end.p0i8(i64 500, i8* nonnull %3)
  53   br label %cleanup
  54 cleanup2:
  55   call void @llvm.lifetime.end.p0i8(i64 300, i8* nonnull %4)
  56   br label %cleanup
  57 cleanup:
  58   call i8* @llvm.coro.free(token %1, i8* %2)
  59   br label %coro.ret
  60 coro.ret:
  61   call i1 @llvm.coro.end(i8* null, i1 false)
  62   ret void
  63 }
  64 ; CHECK:       %a.Frame = type { void (%a.Frame*)*, void (%a.Frame*)*, %"struct.task::promise_type", %struct.big_structure, i1 }
  65 ; CHECK-LABEL: @a.resume(
  66 ; CHECK:         %[[A:.*]] = getelementptr inbounds %a.Frame, %a.Frame* %FramePtr, i32 0, i32 3
  67 ; CHECK:         %{{.*}} = bitcast %struct.big_structure* %[[A]] to %struct.big_structure.2*
  68
  69 declare token @llvm.coro.id(i32, i8* readnone, i8* nocapture readonly, i8*)
  70 declare i1 @llvm.coro.alloc(token) #3
  71 declare i64 @llvm.coro.size.i64() #5
  72 declare i8* @llvm.coro.begin(token, i8* writeonly) #3
  73 declare token @llvm.coro.save(i8*) #3
  74 declare i8* @llvm.coro.frame() #5
  75 declare i8 @llvm.coro.suspend(token, i1) #3
  76 declare i8* @llvm.coro.free(token, i8* nocapture readonly) #2
  77 declare i1 @llvm.coro.end(i8*, i1) #3
  78 declare void @llvm.lifetime.start.p0i8(i64, i8* nocapture) #4
  79 declare void @llvm.lifetime.end.p0i8(i64, i8* nocapture) #4