1; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
2; RUN: opt -function-specialization -deadargelim -force-function-specialization -S < %s | FileCheck %s
3; RUN: opt -function-specialization -func-specialization-max-iters=1 -deadargelim -force-function-specialization -S < %s | FileCheck %s
4; RUN: opt -function-specialization -func-specialization-max-iters=0 -deadargelim -force-function-specialization -S < %s | FileCheck %s --check-prefix=DISABLED
5; RUN: opt -function-specialization -func-specialization-avg-iters-cost=1 -deadargelim -force-function-specialization -S < %s | FileCheck %s
6
7; DISABLED-NOT: @func.1(
8; DISABLED-NOT: @func.2(
9
10define internal i32 @func(i32* %0, i32 %1, void (i32*)* nocapture %2) {
11  %4 = alloca i32, align 4
12  store i32 %1, i32* %4, align 4
13  %5 = load i32, i32* %4, align 4
14  %6 = icmp slt i32 %5, 1
15  br i1 %6, label %14, label %7
16
177:                                                ; preds = %3
18  %8 = load i32, i32* %4, align 4
19  %9 = sext i32 %8 to i64
20  %10 = getelementptr inbounds i32, i32* %0, i64 %9
21  call void %2(i32* %10)
22  %11 = load i32, i32* %4, align 4
23  %12 = add nsw i32 %11, -1
24  %13 = call i32 @func(i32* %0, i32 %12, void (i32*)* %2)
25  br label %14
26
2714:                                               ; preds = %3, %7
28  ret i32 0
29}
30
31define internal void @increment(i32* nocapture %0) {
32  %2 = load i32, i32* %0, align 4
33  %3 = add nsw i32 %2, 1
34  store i32 %3, i32* %0, align 4
35  ret void
36}
37
38define internal void @decrement(i32* nocapture %0) {
39  %2 = load i32, i32* %0, align 4
40  %3 = add nsw i32 %2, -1
41  store i32 %3, i32* %0, align 4
42  ret void
43}
44
45define i32 @main(i32* %0, i32 %1) {
46; CHECK:    [[TMP3:%.*]] = call i32 @func.2(i32* [[TMP0:%.*]], i32 [[TMP1:%.*]])
47  %3 = call i32 @func(i32* %0, i32 %1, void (i32*)* nonnull @increment)
48; CHECK:    [[TMP4:%.*]] = call i32 @func.1(i32* [[TMP0]], i32 [[TMP3]])
49  %4 = call i32 @func(i32* %0, i32 %3, void (i32*)* nonnull @decrement)
50  ret i32 %4
51}
52
53; CHECK: @func.1(
54; CHECK:    [[TMP3:%.*]] = alloca i32, align 4
55; CHECK:    store i32 [[TMP1:%.*]], i32* [[TMP3]], align 4
56; CHECK:    [[TMP4:%.*]] = load i32, i32* [[TMP3]], align 4
57; CHECK:    [[TMP5:%.*]] = icmp slt i32 [[TMP4]], 1
58; CHECK:    br i1 [[TMP5]], label [[TMP13:%.*]], label [[TMP6:%.*]]
59; CHECK:       6:
60; CHECK:    [[TMP7:%.*]] = load i32, i32* [[TMP3]], align 4
61; CHECK:    [[TMP8:%.*]] = sext i32 [[TMP7]] to i64
62; CHECK:    [[TMP9:%.*]] = getelementptr inbounds i32, i32* [[TMP0:%.*]], i64 [[TMP8]]
63; CHECK:    call void @decrement(i32* [[TMP9]])
64; CHECK:    [[TMP10:%.*]] = load i32, i32* [[TMP3]], align 4
65; CHECK:    [[TMP11:%.*]] = add nsw i32 [[TMP10]], -1
66; CHECK:    [[TMP12:%.*]] = call i32 @func.1(i32* [[TMP0]], i32 [[TMP11]])
67; CHECK:    br label [[TMP13]]
68; CHECK:       13:
69; CHECK:    ret i32 0
70;
71;
72; CHECK: @func.2(
73; CHECK:    [[TMP3:%.*]] = alloca i32, align 4
74; CHECK:    store i32 [[TMP1:%.*]], i32* [[TMP3]], align 4
75; CHECK:    [[TMP4:%.*]] = load i32, i32* [[TMP3]], align 4
76; CHECK:    [[TMP5:%.*]] = icmp slt i32 [[TMP4]], 1
77; CHECK:    br i1 [[TMP5]], label [[TMP13:%.*]], label [[TMP6:%.*]]
78; CHECK:       6:
79; CHECK:    [[TMP7:%.*]] = load i32, i32* [[TMP3]], align 4
80; CHECK:    [[TMP8:%.*]] = sext i32 [[TMP7]] to i64
81; CHECK:    [[TMP9:%.*]] = getelementptr inbounds i32, i32* [[TMP0:%.*]], i64 [[TMP8]]
82; CHECK:    call void @increment(i32* [[TMP9]])
83; CHECK:    [[TMP10:%.*]] = load i32, i32* [[TMP3]], align 4
84; CHECK:    [[TMP11:%.*]] = add nsw i32 [[TMP10]], -1
85; CHECK:    [[TMP12:%.*]] = call i32 @func.2(i32* [[TMP0]], i32 [[TMP11]])
86; CHECK:    br label [[TMP13]]
87; CHECK:    ret i32 0
88