| Eric Christopher | cee313d | 2019-04-17 04:52:47 +0000 | [diff] [blame] | 1 | ; RUN: opt < %s -partial-inliner -S -skip-partial-inlining-cost-analysis | FileCheck %s |
| 2 | ; RUN: opt < %s -passes=partial-inliner -S -skip-partial-inlining-cost-analysis | FileCheck %s |
| 3 | |
| 4 | @stat = external global i32, align 4 |
| 5 | |
| 6 | define i32 @vararg(i32 %count, ...) { |
| 7 | entry: |
| 8 | %vargs = alloca i8*, align 8 |
| 9 | %stat1 = load i32, i32* @stat, align 4 |
| 10 | %cmp = icmp slt i32 %stat1, 0 |
| 11 | br i1 %cmp, label %bb2, label %bb1 |
| 12 | |
| 13 | bb1: ; preds = %entry |
| 14 | %vg1 = add nsw i32 %stat1, 1 |
| 15 | store i32 %vg1, i32* @stat, align 4 |
| 16 | %vargs1 = bitcast i8** %vargs to i8* |
| 17 | call void @llvm.va_start(i8* %vargs1) |
| 18 | %va1 = va_arg i8** %vargs, i32 |
| 19 | call void @foo(i32 %count, i32 %va1) #2 |
| 20 | call void @llvm.va_end(i8* %vargs1) |
| 21 | br label %bb2 |
| 22 | |
| 23 | bb2: ; preds = %bb1, %entry |
| 24 | %res = phi i32 [ 1, %bb1 ], [ 0, %entry ] |
| 25 | ret i32 %res |
| 26 | } |
| 27 | |
| 28 | declare void @foo(i32, i32) |
| 29 | declare void @llvm.va_start(i8*) |
| 30 | declare void @llvm.va_end(i8*) |
| 31 | |
| 32 | define i32 @caller1(i32 %arg) { |
| 33 | bb: |
| 34 | %tmp = tail call i32 (i32, ...) @vararg(i32 %arg) |
| 35 | ret i32 %tmp |
| 36 | } |
| 37 | ; CHECK-LABEL: @caller1 |
| 38 | ; CHECK: codeRepl.i: |
| 39 | ; CHECK-NEXT: call void (i32, i8**, i32, ...) @vararg.3.bb1(i32 %stat1.i, i8** %vargs.i, i32 %arg) |
| 40 | |
| 41 | define i32 @caller2(i32 %arg, float %arg2) { |
| 42 | bb: |
| 43 | %tmp = tail call i32 (i32, ...) @vararg(i32 %arg, i32 10, float %arg2) |
| 44 | ret i32 %tmp |
| 45 | } |
| 46 | |
| 47 | ; CHECK-LABEL: @caller2 |
| 48 | ; CHECK: codeRepl.i: |
| 49 | ; CHECK-NEXT: call void (i32, i8**, i32, ...) @vararg.3.bb1(i32 %stat1.i, i8** %vargs.i, i32 %arg, i32 10, float %arg2) |
| 50 | |
| 51 | ; Test case to check that we do not extract a vararg function, if va_end is in |
| 52 | ; a block that is not outlined. |
| 53 | define i32 @vararg_not_legal(i32 %count, ...) { |
| 54 | entry: |
| 55 | %vargs = alloca i8*, align 8 |
| 56 | %vargs0 = bitcast i8** %vargs to i8* |
| 57 | %stat1 = load i32, i32* @stat, align 4 |
| 58 | %cmp = icmp slt i32 %stat1, 0 |
| 59 | br i1 %cmp, label %bb2, label %bb1 |
| 60 | |
| 61 | bb1: ; preds = %entry |
| 62 | %vg1 = add nsw i32 %stat1, 1 |
| 63 | store i32 %vg1, i32* @stat, align 4 |
| 64 | %vargs1 = bitcast i8** %vargs to i8* |
| 65 | call void @llvm.va_start(i8* %vargs1) |
| 66 | %va1 = va_arg i8** %vargs, i32 |
| 67 | call void @foo(i32 %count, i32 %va1) |
| 68 | br label %bb2 |
| 69 | |
| 70 | bb2: ; preds = %bb1, %entry |
| 71 | %res = phi i32 [ 1, %bb1 ], [ 0, %entry ] |
| 72 | %ptr = phi i8* [ %vargs1, %bb1 ], [ %vargs0, %entry] |
| 73 | call void @llvm.va_end(i8* %ptr) |
| 74 | ret i32 %res |
| 75 | } |
| 76 | |
| 77 | ; CHECK-LABEL: @caller3 |
| 78 | ; CHECK: tail call i32 (i32, ...) @vararg_not_legal(i32 %arg, i32 %arg) |
| 79 | define i32 @caller3(i32 %arg) { |
| 80 | bb: |
| 81 | %res = tail call i32 (i32, ...) @vararg_not_legal(i32 %arg, i32 %arg) |
| 82 | ret i32 %res |
| 83 | } |
| 84 | |
| 85 | declare i32* @err(i32*) |
| 86 | |
| 87 | define signext i32 @vararg2(i32 * %l, ...) { |
| 88 | entry: |
| 89 | br i1 undef, label %cleanup, label %cond.end |
| 90 | |
| 91 | cond.end: ; preds = %entry |
| 92 | %call51 = call i32* @err(i32* nonnull %l) |
| 93 | unreachable |
| 94 | |
| 95 | cleanup: ; preds = %entry |
| 96 | ret i32 0 |
| 97 | } |
| 98 | |
| 99 | define i32* @caller_with_signext(i32* %foo) { |
| 100 | entry: |
| 101 | %call1 = tail call signext i32 (i32*, ...) @vararg2(i32* %foo, i32 signext 8) |
| 102 | unreachable |
| 103 | } |
| 104 | |
| 105 | ; CHECK-LABEL: @caller_with_signext |
| 106 | ; CHECK: codeRepl.i: |
| 107 | ; CHECK-NEXT: call void (i32*, ...) @vararg2.1.cond.end(i32* %foo, i32 signext 8) |