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

   1 ; Check that we can handle spills of array allocas
   2 ; RUN: opt < %s -passes='cgscc(coro-split),simplifycfg,early-cse' -reuse-storage-in-coroutine-frame -S | FileCheck %s
   3
   4 %struct.big_structure = type { [500 x i8] }
   5 declare void @consume(%struct.big_structure*)
   6
   7 ; Function Attrs: noinline optnone uwtable
   8 define i8* @f(i1 %cond) "coroutine.presplit"="1" {
   9 entry:
  10   %data = alloca %struct.big_structure, align 1
  11   %data2 = alloca %struct.big_structure, align 1
  12   %id = call token @llvm.coro.id(i32 0, i8* null, i8* null, i8* null)
  13   %size = call i32 @llvm.coro.size.i32()
  14   %alloc = call i8* @malloc(i32 %size)
  15   %hdl = call i8* @llvm.coro.begin(token %id, i8* %alloc)
  16   br i1 %cond, label %then, label %else
  17
  18 then:
  19   %0 = bitcast %struct.big_structure* %data to i8*
  20   call void @llvm.lifetime.start.p0i8(i64 500, i8* nonnull %0)
  21   call void @consume(%struct.big_structure* %data)
  22   %suspend.value = call i8 @llvm.coro.suspend(token none, i1 false)
  23   switch i8 %suspend.value, label %coro.ret [i8 0, label %resume
  24                                              i8 1, label %cleanup1]
  25
  26 resume:
  27   call void @llvm.lifetime.end.p0i8(i64 500, i8* nonnull %0)
  28   br label %cleanup1
  29
  30 cleanup1:
  31   call void @llvm.lifetime.end.p0i8(i64 500, i8* nonnull %0)
  32   br label %cleanup
  33
  34 else:
  35   %1 = bitcast %struct.big_structure* %data2 to i8*
  36   call void @llvm.lifetime.start.p0i8(i64 500, i8* nonnull %1)
  37   call void @consume(%struct.big_structure* %data2)
  38   %suspend.value2 = call i8 @llvm.coro.suspend(token none, i1 false)
  39   switch i8 %suspend.value2, label %coro.ret [i8 0, label %resume2
  40                                               i8 1, label %cleanup2]
  41
  42 resume2:
  43   call void @llvm.lifetime.end.p0i8(i64 500, i8* nonnull %1)
  44   br label %cleanup2
  45
  46 cleanup2:
  47   call void @llvm.lifetime.end.p0i8(i64 500, i8* nonnull %1)
  48   br label %cleanup
  49
  50 cleanup:
  51   %mem = call i8* @llvm.coro.free(token %id, i8* %hdl)
  52   call void @free(i8* %mem)
  53   br label %coro.ret
  54 coro.ret:
  55   call i1 @llvm.coro.end(i8* %hdl, i1 0)
  56   ret i8* %hdl
  57 }
  58
  59 ; CHECK-LABEL: @f(
  60 ; CHECK: call i8* @malloc(i32 520)
  61
  62 declare i8* @llvm.coro.free(token, i8*)
  63 declare i32 @llvm.coro.size.i32()
  64 declare i8  @llvm.coro.suspend(token, i1)
  65 declare void @llvm.coro.resume(i8*)
  66 declare void @llvm.coro.destroy(i8*)
  67
  68 declare token @llvm.coro.id(i32, i8*, i8*, i8*)
  69 declare i1 @llvm.coro.alloc(token)
  70 declare i8* @llvm.coro.begin(token, i8*)
  71 declare i1 @llvm.coro.end(i8*, i1)
  72
  73 declare noalias i8* @malloc(i32)
  74 declare double @print(double)
  75 declare void @free(i8*)
  76
  77 declare void @llvm.lifetime.start.p0i8(i64, i8* nocapture)
  78 declare void @llvm.lifetime.end.p0i8(i64, i8* nocapture)