commit 80457ce5fa868719be4e1e5d9020899eb6d0bf16
author Xinliang David Li <davidxl@google.com> Wed Jun 22 02:12:54 2016 +0000
committer Xinliang David Li <davidxl@google.com> Wed Jun 22 02:12:54 2016 +0000
tree 864882b490b5d0f4fdcae7f5a5695c293759808d
parent 189d6a8834e26332e3718cfeb1a27d8169ef7f50

Add an option to enable MBFI dot viewer for a given function

llvm-svn: 273366

llvm/lib/CodeGen/MachineBlockFrequencyInfo.cpp

diff --git a/llvm/lib/CodeGen/MachineBlockFrequencyInfo.cpp b/llvm/lib/CodeGen/MachineBlockFrequencyInfo.cpp
index d0c6261..8843136 100644
--- a/llvm/lib/CodeGen/MachineBlockFrequencyInfo.cpp
+++ b/llvm/lib/CodeGen/MachineBlockFrequencyInfo.cpp
@@ -42,6 +42,9 @@
                           "integer fractional block frequency representation."),
                clEnumValEnd));
 
+static cl::opt<std::string> ViewMachineBlockFreqFuncName("view-mbfi-func-name",
+                                                         cl::Hidden);
+
 namespace llvm {
 
 template <> struct GraphTraits<MachineBlockFrequencyInfo *> {
@@ -137,7 +140,9 @@
     MBFI.reset(new ImplType);
   MBFI->calculate(F, MBPI, MLI);
 #ifndef NDEBUG
-  if (ViewMachineBlockFreqPropagationDAG != GVDT_None) {
+  if (ViewMachineBlockFreqPropagationDAG != GVDT_None &&
+      (ViewMachineBlockFreqFuncName.empty() ||
+       F.getName().equals(ViewMachineBlockFreqFuncName))) {
     view();
   }
 #endif