Mircea Trofin bb6497ffa6
[BPI] Reuse the AsmWriter's BB naming scheme in BranchProbabilityPrinterPass (#73593)
When using `BranchProbabilityPrinterPass`, if a BB has no name, we get pretty unusable information like `edge -> has probability...` (i.e. we have no idea what the vertices of that edge are).

This patch uses `printAsOperand`, which uses the same naming scheme as `Function::dump`, so for example during debugging sessions, the IR obtained from a function and the names used by `BranchProbabilityPrinterPass` will match.

A shortcoming is that `printAsOperand` will result in the numbering algorithm re-running for every edge and every vertex (when `BranchProbabilityPrinterPass` is run on a function). If, for the given scenario, this is a problem, we can revisit this subsequently.

Another nuance is that the entry basic block will be numbered, which may be slightly confusing when it's anonymous, but it's easily identifiable - the first edge would have it as source (and the number should be easily recognizable)
2023-12-02 13:01:48 -08:00

52 lines
2.1 KiB
LLVM

; RUN: opt < %s -passes='print<branch-prob>' -disable-output 2>&1 | FileCheck %s
declare void @g(i32)
; Check correctness of reported probabilities in case of multiple edges between
; basic blocks. In this case sum of probabilities over all edges should be
; returned by BranchProbabilityInfo::getEdgeProbability.
define void @test1(i32 %x) {
;CHECK: edge %entry -> %return probability is 0x0ccccccd / 0x80000000 = 10.00%
;CHECK: edge %entry -> %bb0 probability is 0x26666666 / 0x80000000 = 30.00%
;CHECK: edge %entry -> %bb0 probability is 0x26666666 / 0x80000000 = 30.00%
;CHECK: edge %entry -> %bb0 probability is 0x26666666 / 0x80000000 = 30.00%
;CHECK: edge %entry -> %bb1 probability is 0x26666666 / 0x80000000 = 30.00%
;CHECK: edge %entry -> %bb1 probability is 0x26666666 / 0x80000000 = 30.00%
;CHECK: edge %entry -> %bb1 probability is 0x26666666 / 0x80000000 = 30.00%
;CHECK: edge %entry -> %bb2 probability is 0x26666666 / 0x80000000 = 30.00%
;CHECK: edge %entry -> %bb2 probability is 0x26666666 / 0x80000000 = 30.00%
;CHECK: edge %entry -> %bb2 probability is 0x26666666 / 0x80000000 = 30.00%
;CHECK: edge %bb0 -> %return probability is 0x80000000 / 0x80000000 = 100.00% [HOT edge]
;CHECK: edge %bb1 -> %return probability is 0x80000000 / 0x80000000 = 100.00% [HOT edge]
;CHECK: edge %bb2 -> %return probability is 0x80000000 / 0x80000000 = 100.00% [HOT edge]
entry:
switch i32 %x, label %return [
i32 0, label %bb0
i32 3, label %bb0
i32 6, label %bb0
i32 1, label %bb1
i32 4, label %bb1
i32 7, label %bb1
i32 2, label %bb2
i32 5, label %bb2
i32 8, label %bb2
]
bb0: ; preds = %entry, %entry, %entry
tail call void @g(i32 0)
br label %return
bb1: ; preds = %entry, %entry, %entry
tail call void @g(i32 1)
br label %return
bb2: ; preds = %entry, %entry, %entry
tail call void @g(i32 2)
br label %return
return: ; preds = %bb2, %bb1, %bb0, %entry
ret void
}