1; RUN: opt -function-specialization -force-function-specialization \
2; RUN:   -func-specialization-max-constants=2 -S < %s | FileCheck %s
3
4; RUN: opt -function-specialization -force-function-specialization \
5; RUN:   -func-specialization-max-constants=1 -S < %s | FileCheck %s --check-prefix=CONST1
6
7target datalayout = "e-m:e-i8:8:32-i16:16:32-i64:64-i128:128-n32:64-S128"
8
9@A = external dso_local constant i32, align 4
10@B = external dso_local constant i32, align 4
11@C = external dso_local constant i32, align 4
12@D = external dso_local constant i32, align 4
13
14define dso_local i32 @bar(i32 %x, i32 %y) {
15entry:
16  %tobool = icmp ne i32 %x, 0
17  br i1 %tobool, label %if.then, label %if.else
18
19if.then:
20  %call = call i32 @foo(i32 %x, i32* @A, i32* @C)
21  br label %return
22
23if.else:
24  %call1 = call i32 @foo(i32 %y, i32* @B, i32* @D)
25  br label %return
26
27return:
28  %retval.0 = phi i32 [ %call, %if.then ], [ %call1, %if.else ]
29  ret i32 %retval.0
30}
31
32define internal i32 @foo(i32 %x, i32* %b, i32* %c) {
33entry:
34  %0 = load i32, i32* %b, align 4
35  %add = add nsw i32 %x, %0
36  %1 = load i32, i32* %c, align 4
37  %add1 = add nsw i32 %add, %1
38  ret i32 %add1
39}
40
41; CONST1-NOT: define internal i32 @foo.1(i32 %x, i32* %b, i32* %c)
42; CONST1-NOT: define internal i32 @foo.2(i32 %x, i32* %b, i32* %c)
43
44; CHECK:        define internal i32 @foo.1(i32 %x, i32* %b, i32* %c) {
45; CHECK-NEXT:   entry:
46; CHECK-NEXT:     %0 = load i32, i32* @A, align 4
47; CHECK-NEXT:     %add = add nsw i32 %x, %0
48; CHECK-NEXT:     %1 = load i32, i32* %c, align 4
49; CHECK-NEXT:     %add1 = add nsw i32 %add, %1
50; CHECK-NEXT:     ret i32 %add1
51; CHECK-NEXT:   }
52
53; CHECK: define internal i32 @foo.2(i32 %x, i32* %b, i32* %c) {
54; CHECK-NEXT:   entry:
55; CHECK-NEXT:     %0 = load i32, i32* @B, align 4
56; CHECK-NEXT:     %add = add nsw i32 %x, %0
57; CHECK-NEXT:     %1 = load i32, i32* %c, align 4
58; CHECK-NEXT:     %add1 = add nsw i32 %add, %1
59; CHECK-NEXT:     ret i32 %add1
60; CHECK-NEXT:   }
61