xref: /llvm-project/llvm/test/Transforms/CalledValuePropagation/simple-arguments.ll (revision 78f2eb8d0f9cf5e8a79b64f6d64deeb487502e38)
1; RUN: opt -passes=called-value-propagation -S < %s | FileCheck %s
2
3target triple = "aarch64"
4
5
6; This test checks that we propagate the functions through arguments and attach
7; !callees metadata to the call. Such metadata can enable optimizations of this
8; code sequence.
9;
10; For example, the code below a illustrates a contrived sort-like algorithm
11; that accepts a pointer to a comparison function. Since the indirect call to
12; the comparison function has only two targets, the call can be promoted to two
13; direct calls using an if-then-else. The loop can then be unswitched and the
14; called functions inlined. This essentially produces two loops, once
15; specialized for each comparison.
16;
17; CHECK:  %tmp3 = call i1 %cmp(ptr %tmp1, ptr %tmp2), !callees ![[MD:[0-9]+]]
18; CHECK: ![[MD]] = !{ptr @ugt, ptr @ule}
19;
20define void @test_argument(ptr %x, i64 %n, i1 %flag) {
21entry:
22  %tmp0 = sub i64 %n, 1
23  br i1 %flag, label %then, label %else
24
25then:
26  call void @arrange_data(ptr %x, i64 %tmp0, ptr @ugt)
27  br label %merge
28
29else:
30  call void @arrange_data(ptr %x, i64 %tmp0, ptr @ule)
31  br label %merge
32
33merge:
34  ret void
35}
36
37define internal void @arrange_data(ptr %x, i64 %n, ptr %cmp) {
38entry:
39  %tmp0 = icmp eq i64 %n, 1
40  br i1 %tmp0, label %merge, label %for.body
41
42for.body:
43  %i = phi i64 [ 0, %entry ], [ %i.next, %cmp.false ]
44  %i.next = add nuw nsw i64 %i, 1
45  %tmp1 = getelementptr inbounds i64, ptr %x, i64 %i
46  %tmp2 = getelementptr inbounds i64, ptr %x, i64 %i.next
47  %tmp3 = call i1 %cmp(ptr %tmp1, ptr %tmp2)
48  br i1 %tmp3, label %cmp.true, label %cmp.false
49
50cmp.true:
51  call void @swap(ptr %tmp1, ptr %tmp2)
52  br label %cmp.false
53
54cmp.false:
55  %cond = icmp slt i64 %i.next, %n
56  br i1 %cond, label %for.body, label %for.end
57
58for.end:
59  %tmp4 = sub i64 %n, 1
60  call void @arrange_data(ptr %x, i64 %tmp4, ptr %cmp)
61  br label %merge
62
63merge:
64  ret void
65}
66
67define internal i1 @ugt(ptr %a, ptr %b) {
68entry:
69  %tmp0 = load i64, ptr %a
70  %tmp1 = load i64, ptr %b
71  %tmp2 = icmp ugt i64 %tmp0, %tmp1
72  ret i1 %tmp2
73}
74
75define internal i1 @ule(ptr %a, ptr %b) {
76entry:
77  %tmp0 = load i64, ptr %a
78  %tmp1 = load i64, ptr %b
79  %tmp2 = icmp ule i64 %tmp0, %tmp1
80  ret i1 %tmp2
81}
82
83declare void @swap(ptr, ptr)
84