llvm/test/Transforms/Coroutines/coro-retcon-resume-values2.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --include-generated-funcs
   2 ; RUN: opt < %s -passes='cgscc(coro-split),simplifycfg,early-cse,simplifycfg,coro-cleanup' -S | FileCheck %s
   3
   4 define i8* @f(i8* %buffer, i32 %n) "coroutine.presplit"="1" {
   5 entry:
   6   %id = call token @llvm.coro.id.retcon(i32 8, i32 4, i8* %buffer, i8* bitcast (i8* (i8*, i32)* @prototype to i8*), i8* bitcast (i8* (i32)* @allocate to i8*), i8* bitcast (void (i8*)* @deallocate to i8*))
   7   %hdl = call i8* @llvm.coro.begin(token %id, i8* null)
   8   %value0 = call i32 (...) @llvm.coro.suspend.retcon.i32()
   9   %sum0 = call i32 @add(i32 %n, i32 %value0)
  10   %value1 = call i32 (...) @llvm.coro.suspend.retcon.i32()
  11   %sum1 = call i32 @add(i32 %sum0, i32 %value0)
  12   %sum2 = call i32 @add(i32 %sum1, i32 %value1)
  13   %value2 = call i32 (...) @llvm.coro.suspend.retcon.i32()
  14   %sum3 = call i32 @add(i32 %sum2, i32 %value0)
  15   %sum4 = call i32 @add(i32 %sum3, i32 %value1)
  16   %sum5 = call i32 @add(i32 %sum4, i32 %value2)
  17   call void @print(i32 %sum5)
  18   call i1 @llvm.coro.end(i8* %hdl, i1 0)
  19   unreachable
  20 }
  21
  22 declare token @llvm.coro.id.retcon(i32, i32, i8*, i8*, i8*, i8*)
  23 declare i8* @llvm.coro.begin(token, i8*)
  24 declare i32 @llvm.coro.suspend.retcon.i32(...)
  25 declare i1 @llvm.coro.end(i8*, i1)
  26 declare i8* @llvm.coro.prepare.retcon(i8*)
  27
  28 declare i8* @prototype(i8*, i32)
  29
  30 declare noalias i8* @allocate(i32 %size)
  31 declare void @deallocate(i8* %ptr)
  32
  33 declare i32 @add(i32, i32)
  34 declare void @print(i32)
  35
  36 ; CHECK-LABEL: @f(
  37 ; CHECK-NEXT:  entry:
  38 ; CHECK-NEXT:    [[TMP0:%.*]] = call i8* @allocate(i32 20)
  39 ; CHECK-NEXT:    [[TMP1:%.*]] = bitcast i8* [[BUFFER:%.*]] to i8**
  40 ; CHECK-NEXT:    store i8* [[TMP0]], i8** [[TMP1]], align 8
  41 ; CHECK-NEXT:    [[FRAMEPTR:%.*]] = bitcast i8* [[TMP0]] to %f.Frame*
  42 ; CHECK-NEXT:    [[N_SPILL_ADDR:%.*]] = getelementptr inbounds [[F_FRAME:%.*]], %f.Frame* [[FRAMEPTR]], i32 0, i32 0
  43 ; CHECK-NEXT:    store i32 [[N:%.*]], i32* [[N_SPILL_ADDR]], align 4
  44 ; CHECK-NEXT:    ret i8* bitcast (i8* (i8*, i32)* @f.resume.0 to i8*)
  45 ;
  46 ;
  47 ; CHECK-LABEL: @f.resume.0(
  48 ; CHECK-NEXT:  entryresume.0:
  49 ; CHECK-NEXT:    [[TMP2:%.*]] = bitcast i8* [[TMP0:%.*]] to %f.Frame**
  50 ; CHECK-NEXT:    [[FRAMEPTR:%.*]] = load %f.Frame*, %f.Frame** [[TMP2]], align 8
  51 ; CHECK-NEXT:    [[VALUE0_SPILL_ADDR:%.*]] = getelementptr inbounds [[F_FRAME:%.*]], %f.Frame* [[FRAMEPTR]], i32 0, i32 1
  52 ; CHECK-NEXT:    store i32 [[TMP1:%.*]], i32* [[VALUE0_SPILL_ADDR]], align 4
  53 ; CHECK-NEXT:    [[N_RELOAD_ADDR:%.*]] = getelementptr inbounds [[F_FRAME]], %f.Frame* [[FRAMEPTR]], i32 0, i32 0
  54 ; CHECK-NEXT:    [[N_RELOAD:%.*]] = load i32, i32* [[N_RELOAD_ADDR]], align 4
  55 ; CHECK-NEXT:    [[SUM0:%.*]] = call i32 @add(i32 [[N_RELOAD]], i32 [[TMP1]])
  56 ; CHECK-NEXT:    [[SUM0_SPILL_ADDR:%.*]] = getelementptr inbounds [[F_FRAME]], %f.Frame* [[FRAMEPTR]], i32 0, i32 2
  57 ; CHECK-NEXT:    store i32 [[SUM0]], i32* [[SUM0_SPILL_ADDR]], align 4
  58 ; CHECK-NEXT:    ret i8* bitcast (i8* (i8*, i32)* @f.resume.1 to i8*)
  59 ;
  60 ;
  61 ; CHECK-LABEL: @f.resume.1(
  62 ; CHECK-NEXT:  entryresume.1:
  63 ; CHECK-NEXT:    [[TMP2:%.*]] = bitcast i8* [[TMP0:%.*]] to %f.Frame**
  64 ; CHECK-NEXT:    [[FRAMEPTR:%.*]] = load %f.Frame*, %f.Frame** [[TMP2]], align 8
  65 ; CHECK-NEXT:    [[VALUE1_SPILL_ADDR:%.*]] = getelementptr inbounds [[F_FRAME:%.*]], %f.Frame* [[FRAMEPTR]], i32 0, i32 3
  66 ; CHECK-NEXT:    store i32 [[TMP1:%.*]], i32* [[VALUE1_SPILL_ADDR]], align 4
  67 ; CHECK-NEXT:    [[SUM0_RELOAD_ADDR:%.*]] = getelementptr inbounds [[F_FRAME]], %f.Frame* [[FRAMEPTR]], i32 0, i32 2
  68 ; CHECK-NEXT:    [[SUM0_RELOAD:%.*]] = load i32, i32* [[SUM0_RELOAD_ADDR]], align 4
  69 ; CHECK-NEXT:    [[VALUE0_RELOAD_ADDR5:%.*]] = getelementptr inbounds [[F_FRAME]], %f.Frame* [[FRAMEPTR]], i32 0, i32 1
  70 ; CHECK-NEXT:    [[VALUE0_RELOAD6:%.*]] = load i32, i32* [[VALUE0_RELOAD_ADDR5]], align 4
  71 ; CHECK-NEXT:    [[SUM1:%.*]] = call i32 @add(i32 [[SUM0_RELOAD]], i32 [[VALUE0_RELOAD6]])
  72 ; CHECK-NEXT:    [[SUM2:%.*]] = call i32 @add(i32 [[SUM1]], i32 [[TMP1]])
  73 ; CHECK-NEXT:    [[SUM2_SPILL_ADDR:%.*]] = getelementptr inbounds [[F_FRAME]], %f.Frame* [[FRAMEPTR]], i32 0, i32 4
  74 ; CHECK-NEXT:    store i32 [[SUM2]], i32* [[SUM2_SPILL_ADDR]], align 4
  75 ; CHECK-NEXT:    ret i8* bitcast (i8* (i8*, i32)* @f.resume.2 to i8*)
  76 ;
  77 ;
  78 ; CHECK-LABEL: @f.resume.2(
  79 ; CHECK-NEXT:  entryresume.2:
  80 ; CHECK-NEXT:    [[TMP2:%.*]] = bitcast i8* [[TMP0:%.*]] to %f.Frame**
  81 ; CHECK-NEXT:    [[FRAMEPTR:%.*]] = load %f.Frame*, %f.Frame** [[TMP2]], align 8
  82 ; CHECK-NEXT:    [[SUM2_RELOAD_ADDR:%.*]] = getelementptr inbounds [[F_FRAME:%.*]], %f.Frame* [[FRAMEPTR]], i32 0, i32 4
  83 ; CHECK-NEXT:    [[SUM2_RELOAD:%.*]] = load i32, i32* [[SUM2_RELOAD_ADDR]], align 4
  84 ; CHECK-NEXT:    [[VALUE1_RELOAD_ADDR:%.*]] = getelementptr inbounds [[F_FRAME]], %f.Frame* [[FRAMEPTR]], i32 0, i32 3
  85 ; CHECK-NEXT:    [[VALUE1_RELOAD:%.*]] = load i32, i32* [[VALUE1_RELOAD_ADDR]], align 4
  86 ; CHECK-NEXT:    [[VALUE0_RELOAD_ADDR:%.*]] = getelementptr inbounds [[F_FRAME]], %f.Frame* [[FRAMEPTR]], i32 0, i32 1
  87 ; CHECK-NEXT:    [[VALUE0_RELOAD:%.*]] = load i32, i32* [[VALUE0_RELOAD_ADDR]], align 4
  88 ; CHECK-NEXT:    [[SUM3:%.*]] = call i32 @add(i32 [[SUM2_RELOAD]], i32 [[VALUE0_RELOAD]])
  89 ; CHECK-NEXT:    [[SUM4:%.*]] = call i32 @add(i32 [[SUM3]], i32 [[VALUE1_RELOAD]])
  90 ; CHECK-NEXT:    [[SUM5:%.*]] = call i32 @add(i32 [[SUM4]], i32 [[TMP1:%.*]])
  91 ; CHECK-NEXT:    call void @print(i32 [[SUM5]])
  92 ; CHECK-NEXT:    [[TMP3:%.*]] = bitcast %f.Frame* [[FRAMEPTR]] to i8*
  93 ; CHECK-NEXT:    call void @deallocate(i8* [[TMP3]])
  94 ; CHECK-NEXT:    ret i8* null
  95 ;