1; RUN: opt -function-specialization -func-specialization-size-threshold=3 -S < %s | FileCheck %s
2
3define i64 @main(i64 %x, i1 %flag) {
4;
5; CHECK-LABEL: @main(i64 %x, i1 %flag) {
6; CHECK:      entry:
7; CHECK-NEXT:   br i1 %flag, label %plus, label %minus
8; CHECK:      plus:
9; CHECK-NEXT:   [[TMP0:%.+]] = call i64 @compute.1(i64 %x, i64 (i64)* @plus)
10; CHECK-NEXT:   br label %merge
11; CHECK:      minus:
12; CHECK-NEXT:   [[TMP1:%.+]] = call i64 @compute.2(i64 %x, i64 (i64)* @minus)
13; CHECK-NEXT:   br label %merge
14; CHECK:      merge:
15; CHECK-NEXT:   [[TMP2:%.+]] = phi i64 [ [[TMP0]], %plus ], [ [[TMP1]], %minus ]
16; CHECK-NEXT:   ret i64 [[TMP2]]
17; CHECK-NEXT: }
18;
19entry:
20  br i1 %flag, label %plus, label %minus
21
22plus:
23  %tmp0 = call i64 @compute(i64 %x, i64 (i64)* @plus)
24  br label %merge
25
26minus:
27  %tmp1 = call i64 @compute(i64 %x, i64 (i64)* @minus)
28  br label %merge
29
30merge:
31  %tmp2 = phi i64 [ %tmp0, %plus ], [ %tmp1, %minus]
32  ret i64 %tmp2
33}
34
35; CHECK-NOT: define internal i64 @compute(
36;
37; CHECK-LABEL: define internal i64 @compute.1(i64 %x, i64 (i64)* %binop) {
38; CHECK-NEXT:  entry:
39; CHECK-NEXT:    [[TMP0:%.+]] = call i64 @plus(i64 %x)
40; CHECK-NEXT:    ret i64 [[TMP0]]
41; CHECK-NEXT:  }
42;
43; CHECK-LABEL: define internal i64 @compute.2(i64 %x, i64 (i64)* %binop) {
44; CHECK-NEXT:  entry:
45; CHECK-NEXT:    [[TMP0:%.+]] = call i64 @minus(i64 %x)
46; CHECK-NEXT:    ret i64 [[TMP0]]
47; CHECK-NEXT:  }
48;
49define internal i64 @compute(i64 %x, i64 (i64)* %binop) {
50entry:
51  %tmp0 = call i64 %binop(i64 %x)
52  ret i64 %tmp0
53}
54
55define internal i64 @plus(i64 %x) {
56entry:
57  %tmp0 = add i64 %x, 1
58  ret i64 %tmp0
59}
60
61define internal i64 @minus(i64 %x) {
62entry:
63  %tmp0 = sub i64 %x, 1
64  ret i64 %tmp0
65}
66