xref: /llvm-project/llvm/test/Transforms/FunctionSpecialization/function-specialization3.ll (revision e15d72adac66790a78a51a3087b56b2bde6b778a)
1; RUN: opt -passes="ipsccp<func-spec>" -S < %s | \
2; RUN:   FileCheck %s --check-prefixes=COMMON,DISABLED
3; RUN: opt -passes="ipsccp<func-spec>" -force-specialization -S < %s | \
4; RUN:   FileCheck %s --check-prefixes=COMMON,FORCE
5
6; Test for specializing a constant global.
7
8target datalayout = "e-m:e-i8:8:32-i16:16:32-i64:64-i128:128-n32:64-S128"
9
10@A = external dso_local constant i32, align 4
11@B = external dso_local constant i32, align 4
12
13define dso_local i32 @bar(i32 %x, i32 %y) {
14; COMMON-LABEL: @bar
15; FORCE:        %call = call i32 @foo.specialized.1(i32 %x, ptr @A)
16; FORCE:        %call1 = call i32 @foo.specialized.2(i32 %y, ptr @B)
17; DISABLED-NOT: %call1 = call i32 @foo.specialized.1(
18entry:
19  %tobool = icmp ne i32 %x, 0
20  br i1 %tobool, label %if.then, label %if.else
21
22if.then:
23  %call = call i32 @foo(i32 %x, ptr @A)
24  br label %return
25
26if.else:
27  %call1 = call i32 @foo(i32 %y, ptr @B)
28  br label %return
29
30return:
31  %retval.0 = phi i32 [ %call, %if.then ], [ %call1, %if.else ]
32  ret i32 %retval.0
33}
34
35; FORCE-NOT: define internal i32 @foo(
36;
37; FORCE:      define internal i32 @foo.specialized.1(i32 %x, ptr %b) {
38; FORCE-NEXT: entry:
39; FORCE-NEXT:   %0 = load i32, ptr @A, align 4
40; FORCE-NEXT:   %add = add nsw i32 %x, %0
41; FORCE-NEXT:   ret i32 %add
42; FORCE-NEXT: }
43;
44; FORCE:      define internal i32 @foo.specialized.2(i32 %x, ptr %b) {
45; FORCE-NEXT: entry:
46; FORCE-NEXT:   %0 = load i32, ptr @B, align 4
47; FORCE-NEXT:   %add = add nsw i32 %x, %0
48; FORCE-NEXT:   ret i32 %add
49; FORCE-NEXT: }
50;
51define internal i32 @foo(i32 %x, ptr %b) {
52entry:
53  %0 = load i32, ptr %b, align 4
54  %add = add nsw i32 %x, %0
55  ret i32 %add
56}
57