[llvm] [profcheck] Add indirect call metadata (PR #154657)

Mingming Liu via llvm-commits llvm-commits at lists.llvm.org
Wed Aug 20 18:51:44 PDT 2025


================
@@ -92,12 +97,16 @@ bool ProfileInjector::inject() {
     return false;
   bool Changed = false;
   for (auto &BB : F) {
-    if (AnnotateSelect) {
-      for (auto &I : BB) {
-        if (isa<SelectInst>(I) && !I.getMetadata(LLVMContext::MD_prof))
-          setBranchWeights(I, {SelectTrueWeight, SelectFalseWeight},
-                           /*IsExpected=*/false);
-      }
+    for (auto &I : BB) {
+      if (AnnotateSelect && isa<SelectInst>(I) &&
+          !I.getMetadata(LLVMContext::MD_prof))
+        setBranchWeights(I, {SelectTrueWeight, SelectFalseWeight},
+                         /*IsExpected=*/false);
+      if (AnnotateIndirectCalls)
+        if (auto *CB = dyn_cast<CallBase>(&I))
+          if (CB->isIndirectCall() && !CB->getMetadata(LLVMContext::MD_prof))
+            annotateValueSite(*F.getParent(), *CB, {{2345, 10}, {5678, 20}}, 30,
----------------
mingmingl-llvm wrote:

nit: add some comments for the numbers 

https://github.com/llvm/llvm-project/pull/154657


More information about the llvm-commits mailing list