[llvm] Optionally print `!prof` metadata inline (PR #130303)

[Mircea Trofin via llvm-commits]

https://github.com/mtrofin updated https://github.com/llvm/llvm-project/pull/130303

>From 2d5a8813f77857d1ce025f7bf10534c0e337abf2 Mon Sep 17 00:00:00 2001
From: Mircea Trofin <mtrofin at google.com>
Date: Fri, 7 Mar 2025 08:19:31 -0800
Subject: [PATCH] Optionally print `!prof` metadata inline

---
 llvm/lib/IR/AsmWriter.cpp          | 18 ++++++++++++++++++
 llvm/test/Other/print-prof-data.ll | 24 ++++++++++++++++++++++++
 2 files changed, 42 insertions(+)
 create mode 100644 llvm/test/Other/print-prof-data.ll

diff --git a/llvm/lib/IR/AsmWriter.cpp b/llvm/lib/IR/AsmWriter.cpp
index 88bb2359cd055..1addb7ca97df0 100644
--- a/llvm/lib/IR/AsmWriter.cpp
+++ b/llvm/lib/IR/AsmWriter.cpp
@@ -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> PrintPerfData(
+    "print-perf-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,12 @@ void AssemblyWriter::printFunction(const Function *F) {
     writeOperand(F->getPersonalityFn(), /*PrintType=*/true);
   }
 
+  if (PrintPerfData)
+    if (auto *MDProf = F->getMetadata(LLVMContext::MD_prof)) {
+      Out << " ";
+      MDProf->print(Out, TheModule, /*IsForDebug=*/true);
+    }
+
   if (F->isDeclaration()) {
     Out << '\n';
   } else {
@@ -4287,6 +4297,14 @@ void AssemblyWriter::printInfoComment(const Value &V) {
       }
     }
   }
+  if (PrintPerfData) {
+    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;
diff --git a/llvm/test/Other/print-prof-data.ll b/llvm/test/Other/print-prof-data.ll
new file mode 100644
index 0000000000000..a5a2d444222b5
--- /dev/null
+++ b/llvm/test/Other/print-prof-data.ll
@@ -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}
\ No newline at end of file