aboutsummaryrefslogtreecommitdiff
path: root/llvm/test/Transforms/CodeExtractor/PartialInlineVarArg.ll
blob: 7589db93e42ba402d03fb04d9cee8db6f6b2af79 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
; RUN: opt < %s -passes=partial-inliner -S -skip-partial-inlining-cost-analysis | FileCheck %s

@stat = external global i32, align 4

define i32 @vararg(i32 %count, ...) {
entry:
  %vargs = alloca ptr, align 8
  %stat1 = load i32, ptr @stat, align 4
  %cmp = icmp slt i32 %stat1, 0
  br i1 %cmp, label %bb2, label %bb1

bb1:                                              ; preds = %entry
  %vg1 = add nsw i32 %stat1, 1
  store i32 %vg1, ptr @stat, align 4
  call void @llvm.va_start(ptr %vargs)
  %va1 = va_arg ptr %vargs, i32
  call void @foo(i32 %count, i32 %va1) #2
  call void @llvm.va_end(ptr %vargs)
  br label %bb2

bb2:                                              ; preds = %bb1, %entry
  %res = phi i32 [ 1, %bb1 ], [ 0, %entry ]
  ret i32 %res
}

declare void @foo(i32, i32)
declare void @llvm.va_start(ptr)
declare void @llvm.va_end(ptr)

define i32 @caller1(i32 %arg) {
bb:
  %tmp = tail call i32 (i32, ...) @vararg(i32 %arg)
  ret i32 %tmp
}
; CHECK-LABEL: @caller1
; CHECK: codeRepl.i:
; CHECK-NEXT:  call void (i32, ptr, i32, ...) @vararg.3.bb1(i32 %stat1.i, ptr %vargs.i, i32 %arg)

define i32 @caller2(i32 %arg, float %arg2) {
bb:
  %tmp = tail call i32 (i32, ...) @vararg(i32 %arg, i32 10, float %arg2)
  ret i32 %tmp
}

; CHECK-LABEL: @caller2
; CHECK: codeRepl.i:
; CHECK-NEXT:  call void (i32, ptr, i32, ...) @vararg.3.bb1(i32 %stat1.i, ptr %vargs.i, i32 %arg, i32 10, float %arg2)

; Test case to check that we do not extract a vararg function, if va_end is in
; a block that is not outlined.
define i32 @vararg_not_legal(i32 %count, ...) {
entry:
  %vargs = alloca ptr, align 8
  %stat1 = load i32, ptr @stat, align 4
  %cmp = icmp slt i32 %stat1, 0
  br i1 %cmp, label %bb2, label %bb1

bb1:                                              ; preds = %entry
  %vg1 = add nsw i32 %stat1, 1
  store i32 %vg1, ptr @stat, align 4
  call void @llvm.va_start(ptr %vargs)
  %va1 = va_arg ptr %vargs, i32
  call void @foo(i32 %count, i32 %va1)
  br label %bb2

bb2:                                              ; preds = %bb1, %entry
  %res = phi i32 [ 1, %bb1 ], [ 0, %entry ]
  %ptr = phi ptr [ %vargs, %bb1 ], [ %vargs, %entry]
  call void @llvm.va_end(ptr %ptr)
  ret i32 %res
}

; CHECK-LABEL: @caller3
; CHECK: tail call i32 (i32, ...) @vararg_not_legal(i32 %arg, i32 %arg)
define i32 @caller3(i32 %arg) {
bb:
  %res = tail call i32 (i32, ...) @vararg_not_legal(i32 %arg, i32 %arg)
  ret i32 %res
}

declare ptr @err(ptr)

define signext i32 @vararg2(ptr %l, ...) {
entry:
  %c = load i1, ptr %l
  br i1 %c, label %cleanup, label %cond.end

cond.end:                                         ; preds = %entry
  %call51 = call ptr @err(ptr nonnull %l)
  unreachable

cleanup:                                          ; preds = %entry
  ret i32 0
}

define ptr @caller_with_signext(ptr %foo) {
entry:
  %call1 = tail call signext i32 (ptr, ...) @vararg2(ptr %foo, i32 signext 8)
  unreachable
}

; CHECK-LABEL: @caller_with_signext
; CHECK: codeRepl.i:
; CHECK-NEXT:  call void (ptr, ...) @vararg2.1.cond.end(ptr %foo, i32 signext 8)