xref: /aosp_15_r20/external/llvm/test/CodeGen/Generic/MachineBranchProb.ll (revision 9880d6810fe72a1726cb53787c6711e909410d58)
1*9880d681SAndroid Build Coastguard Worker; RUN: llc < %s -print-machineinstrs=expand-isel-pseudos -o /dev/null 2>&1 | FileCheck %s
2*9880d681SAndroid Build Coastguard Worker
3*9880d681SAndroid Build Coastguard Worker; ARM & AArch64 run an extra SimplifyCFG which disrupts this test.
4*9880d681SAndroid Build Coastguard Worker; Hexagon crashes (PR23377)
5*9880d681SAndroid Build Coastguard Worker; XFAIL: arm,aarch64
6*9880d681SAndroid Build Coastguard Worker
7*9880d681SAndroid Build Coastguard Worker; Make sure we have the correct weight attached to each successor.
8*9880d681SAndroid Build Coastguard Workerdefine i32 @test2(i32 %x) nounwind uwtable readnone ssp {
9*9880d681SAndroid Build Coastguard Worker; CHECK-LABEL: Machine code for function test2:
10*9880d681SAndroid Build Coastguard Workerentry:
11*9880d681SAndroid Build Coastguard Worker  %conv = sext i32 %x to i64
12*9880d681SAndroid Build Coastguard Worker  switch i64 %conv, label %return [
13*9880d681SAndroid Build Coastguard Worker    i64 0, label %sw.bb
14*9880d681SAndroid Build Coastguard Worker    i64 1, label %sw.bb
15*9880d681SAndroid Build Coastguard Worker    i64 4, label %sw.bb
16*9880d681SAndroid Build Coastguard Worker    i64 5, label %sw.bb1
17*9880d681SAndroid Build Coastguard Worker  ], !prof !0
18*9880d681SAndroid Build Coastguard Worker; CHECK: BB#0: derived from LLVM BB %entry
19*9880d681SAndroid Build Coastguard Worker; CHECK: Successors according to CFG: BB#2({{[0-9a-fx/= ]+}}75.29%) BB#4({{[0-9a-fx/= ]+}}24.71%)
20*9880d681SAndroid Build Coastguard Worker; CHECK: BB#4: derived from LLVM BB %entry
21*9880d681SAndroid Build Coastguard Worker; CHECK: Successors according to CFG: BB#1({{[0-9a-fx/= ]+}}47.62%) BB#5({{[0-9a-fx/= ]+}}52.38%)
22*9880d681SAndroid Build Coastguard Worker; CHECK: BB#5: derived from LLVM BB %entry
23*9880d681SAndroid Build Coastguard Worker; CHECK: Successors according to CFG: BB#1({{[0-9a-fx/= ]+}}36.36%) BB#3({{[0-9a-fx/= ]+}}63.64%)
24*9880d681SAndroid Build Coastguard Worker
25*9880d681SAndroid Build Coastguard Workersw.bb:
26*9880d681SAndroid Build Coastguard Worker  br label %return
27*9880d681SAndroid Build Coastguard Worker
28*9880d681SAndroid Build Coastguard Workersw.bb1:
29*9880d681SAndroid Build Coastguard Worker  br label %return
30*9880d681SAndroid Build Coastguard Worker
31*9880d681SAndroid Build Coastguard Workerreturn:
32*9880d681SAndroid Build Coastguard Worker  %retval.0 = phi i32 [ 5, %sw.bb1 ], [ 1, %sw.bb ], [ 0, %entry ]
33*9880d681SAndroid Build Coastguard Worker  ret i32 %retval.0
34*9880d681SAndroid Build Coastguard Worker}
35*9880d681SAndroid Build Coastguard Worker
36*9880d681SAndroid Build Coastguard Worker!0 = !{!"branch_weights", i32 7, i32 6, i32 4, i32 4, i32 64}
37*9880d681SAndroid Build Coastguard Worker
38*9880d681SAndroid Build Coastguard Worker
39*9880d681SAndroid Build Coastguard Workerdeclare void @g(i32)
40*9880d681SAndroid Build Coastguard Workerdefine void @left_leaning_weight_balanced_tree(i32 %x) {
41*9880d681SAndroid Build Coastguard Workerentry:
42*9880d681SAndroid Build Coastguard Worker  switch i32 %x, label %return [
43*9880d681SAndroid Build Coastguard Worker    i32 0,  label %bb0
44*9880d681SAndroid Build Coastguard Worker    i32 100, label %bb1
45*9880d681SAndroid Build Coastguard Worker    i32 200, label %bb2
46*9880d681SAndroid Build Coastguard Worker    i32 300, label %bb3
47*9880d681SAndroid Build Coastguard Worker    i32 400, label %bb4
48*9880d681SAndroid Build Coastguard Worker    i32 500, label %bb5
49*9880d681SAndroid Build Coastguard Worker  ], !prof !1
50*9880d681SAndroid Build Coastguard Workerbb0: tail call void @g(i32 0) br label %return
51*9880d681SAndroid Build Coastguard Workerbb1: tail call void @g(i32 1) br label %return
52*9880d681SAndroid Build Coastguard Workerbb2: tail call void @g(i32 2) br label %return
53*9880d681SAndroid Build Coastguard Workerbb3: tail call void @g(i32 3) br label %return
54*9880d681SAndroid Build Coastguard Workerbb4: tail call void @g(i32 4) br label %return
55*9880d681SAndroid Build Coastguard Workerbb5: tail call void @g(i32 5) br label %return
56*9880d681SAndroid Build Coastguard Workerreturn: ret void
57*9880d681SAndroid Build Coastguard Worker
58*9880d681SAndroid Build Coastguard Worker; Check that we set branch weights on the pivot cmp instruction correctly.
59*9880d681SAndroid Build Coastguard Worker; Cases {0,10,20,30} go on the left with weight 13; cases {40,50} go on the
60*9880d681SAndroid Build Coastguard Worker; right with weight 20.
61*9880d681SAndroid Build Coastguard Worker;
62*9880d681SAndroid Build Coastguard Worker; CHECK-LABEL: Machine code for function left_leaning_weight_balanced_tree:
63*9880d681SAndroid Build Coastguard Worker; CHECK: BB#0: derived from LLVM BB %entry
64*9880d681SAndroid Build Coastguard Worker; CHECK-NOT: Successors
65*9880d681SAndroid Build Coastguard Worker; CHECK: Successors according to CFG: BB#8({{[0-9a-fx/= ]+}}39.71%) BB#9({{[0-9a-fx/= ]+}}60.29%)
66*9880d681SAndroid Build Coastguard Worker}
67*9880d681SAndroid Build Coastguard Worker
68*9880d681SAndroid Build Coastguard Worker!1 = !{!"branch_weights",
69*9880d681SAndroid Build Coastguard Worker  ; Default:
70*9880d681SAndroid Build Coastguard Worker  i32 1,
71*9880d681SAndroid Build Coastguard Worker  ; Case 0, 100, 200:
72*9880d681SAndroid Build Coastguard Worker  i32 10, i32 1, i32 1,
73*9880d681SAndroid Build Coastguard Worker  ; Case 300, 400, 500:
74*9880d681SAndroid Build Coastguard Worker  i32 1, i32 10, i32 10}
75