xref: /llvm-project/llvm/test/Transforms/PGOProfile/split-indirectbr-critical-edges.ll (revision 9ff36df5a4a7d52c51e950522870bb64912688d2)
1; RUN: opt < %s -passes=pgo-instr-gen -S | FileCheck %s
2
3; Function Attrs: norecurse nounwind readnone uwtable
4define i32 @bar(i32 %v) local_unnamed_addr #0 {
5entry:
6  %mul = shl nsw i32 %v, 1
7  ret i32 %mul
8}
9
10; Function Attrs: norecurse nounwind readonly uwtable
11define i32 @foo(ptr nocapture readonly %p) #1 {
12entry:
13  %targets = alloca [256 x ptr], align 16
14  %arrayidx1 = getelementptr inbounds [256 x ptr], ptr %targets, i64 0, i64 93
15  store ptr blockaddress(@foo, %if.end), ptr %arrayidx1, align 8
16  br label %for.cond2
17
18for.cond2:                                        ; preds = %if.end, %for.cond2, %entry
19; CHECK: for.cond2:                                        ; preds = %.split1
20  %p.addr.0 = phi ptr [ %p, %entry ], [ %incdec.ptr5, %if.end ], [ %incdec.ptr, %for.cond2 ]
21  %incdec.ptr = getelementptr inbounds i8, ptr %p.addr.0, i64 1
22  %0 = load i8, ptr %p.addr.0, align 1
23  %cond = icmp eq i8 %0, 93
24  br i1 %cond, label %if.end.preheader, label %for.cond2
25
26if.end.preheader:                                 ; preds = %for.cond2
27  br label %if.end
28
29if.end:                                           ; preds = %if.end.preheader, %if.end
30; CHECK: if.end:                                           ; preds = %.split1
31  %p.addr.1 = phi ptr [ %incdec.ptr5, %if.end ], [ %incdec.ptr, %if.end.preheader ]
32  %incdec.ptr5 = getelementptr inbounds i8, ptr %p.addr.1, i64 1
33  %1 = load i8, ptr %p.addr.1, align 1
34  %idxprom6 = zext i8 %1 to i64
35  %arrayidx7 = getelementptr inbounds [256 x ptr], ptr %targets, i64 0, i64 %idxprom6
36  %2 = load ptr, ptr %arrayidx7, align 8
37  indirectbr ptr %2, [label %for.cond2, label %if.end]
38; CHECK: indirectbr ptr %2, [label %for.cond2, label %if.end]
39}
40
41;; If an indirectbr critical edge cannot be split, ignore it.
42;; The edge will not be profiled.
43; CHECK-LABEL: @cannot_split(
44; CHECK-NEXT:  entry:
45; CHECK-NEXT:    call void @llvm.instrprof.increment
46; CHECK: indirect:
47; CHECK-NOT:     call void @llvm.instrprof.increment
48; CHECK: indirect2:
49; CHECK-NEXT:    call void @llvm.instrprof.increment
50define i32 @cannot_split(ptr nocapture readonly %p) {
51entry:
52  %targets = alloca <2 x ptr>, align 16
53  store <2 x ptr> <ptr blockaddress(@cannot_split, %indirect), ptr blockaddress(@cannot_split, %end)>, ptr %targets, align 16
54  %arrayidx2 = getelementptr inbounds i8, ptr %p, i64 1
55  %0 = load i8, ptr %arrayidx2
56  %idxprom = sext i8 %0 to i64
57  %arrayidx3 = getelementptr inbounds <2 x ptr>, ptr %targets, i64 0, i64 %idxprom
58  %1 = load ptr, ptr %arrayidx3, align 8
59  br label %indirect
60
61indirect:                                         ; preds = %entry, %indirect
62  indirectbr ptr %1, [label %indirect, label %end, label %indirect2]
63
64indirect2:
65  ; For this test we do not want critical edges split. Adding a 2nd `indirectbr`
66  ; does the trick.
67  indirectbr ptr %1, [label %indirect, label %end]
68
69end:                                              ; preds = %indirect
70  ret i32 0
71}
72