mirrors/llvm-project
0
0
Fork 0
mirror of https://github.com/llvm/llvm-project.git synced 2025-04-18 23:56:49 +00:00
Code Issues Projects Releases Wiki Activity

Optionally print !prof metadata inline (#130303)

Browse Source 
Inspired by PR #127944, this patch adds an option to print profile metadata inline with respect to the instruction (or function) it annotates - this saves one time from having to search up and down large textual modules to find this info.
This commit is contained in:
Mircea Trofin 2025-03-07 12:22:13 -08:00 committed by GitHub
parent 0ea02e7721
commit 11b1f154be
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 43 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
llvm/lib/IR/AsmWriter.cpp
View File

@ -96,6 +96,10 @@ static cl::opt<bool> PrintInstDebugLocs(
"print-inst-debug-locs", cl::Hidden,
cl::desc("Pretty print debug locations of instructions when dumping"));
static cl::opt<bool> PrintProfData(
"print-prof-data", cl::Hidden,
cl::desc("Pretty print perf data (branch weights, etc) when dumping"));
// Make virtual table appear in this compilation unit.
AssemblyAnnotationWriter::~AssemblyAnnotationWriter() = default;
@ -4161,6 +4165,13 @@ void AssemblyWriter::printFunction(const Function *F) {
writeOperand(F->getPersonalityFn(), /*PrintType=*/true);
}
if (PrintProfData) {
if (auto *MDProf = F->getMetadata(LLVMContext::MD_prof)) {
Out << " ";
MDProf->print(Out, TheModule, /*IsForDebug=*/true);
}
}
if (F->isDeclaration()) {
Out << '\n';
} else {
@ -4287,6 +4298,14 @@ void AssemblyWriter::printInfoComment(const Value &V) {
}
}
}
if (PrintProfData) {
if (auto *I = dyn_cast<Instruction>(&V)) {
if (auto *MD = I->getMetadata(LLVMContext::MD_prof)) {
Out << " ; ";
MD->print(Out, TheModule, /*IsForDebug=*/true);
}
}
}
if (PrintInstAddrs)
Out << " ; " << &V;

24
llvm/test/Other/print-prof-data.ll Normal file
View File

@ -0,0 +1,24 @@
; RUN: opt %s -print-prof-data -S | FileCheck %s
define void @foo(ptr %p) !prof !0 {
%isnull = icmp eq ptr %p, null
br i1 %isnull, label %yes, label %no, !prof !1
yes:
%something = select i1 %isnull, i32 1, i32 2, !prof !2
br label %exit
no:
call void %p(), !prof !3
br label %exit
exit:
ret void
}
!0 = !{!"function_entry_count", i64 42}
!1 = !{!"branch_weights", i64 20, i64 101}
!2 = !{!"branch_weights", i64 5, i64 70}
!3 = !{!"VP", i32 0, i64 4, i64 4445083295448962937, i64 2, i64 -2718743882639408571, i64 2}
; CHECK: define void @foo(ptr %p) !0 = !{!"function_entry_count", i64 42} !prof !0 {
; CHECK: br i1 %isnull, label %yes, label %no, !prof !1 ; !1 = !{!"branch_weights", i64 20, i64 101}
; CHECK: %something = select i1 %isnull, i32 1, i32 2, !prof !2 ; !2 = !{!"branch_weights", i64 5, i64 70}
; CHECK: call void %p(), !prof !3 ; !3 = !{!"VP", i32 0, i64 4, i64 4445083295448962937, i64 2, i64 -2718743882639408571, i64 2}