Print the right call set size

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@4688 91177308-0d34-0410-b5e6-96231b3b80d8

1 files changed, 3 insertions, 1 deletions

diff --git a/lib/Analysis/DataStructure/Printer.cpp b/lib/Analysis/DataStructure/Printer.cpp
index 483d223535..8b6bdcf6dc 100644
--- a/lib/Analysis/DataStructure/Printer.cpp
+++ b/lib/Analysis/DataStructure/Printer.cpp
@@ -155,7 +155,9 @@ void DSGraph::writeGraphToFile(std::ostream &O, const string &GraphName) const {
   
   if (F.good()) {
     print(F);
-    O << " [" << getGraphSize() << "+" << getFunctionCalls().size() << "]\n";
+    unsigned NumCalls = shouldPrintAuxCalls() ?
+      getAuxFunctionCalls().size() : getFunctionCalls().size();
+    O << " [" << getGraphSize() << "+" << NumCalls << "]\n";
   } else {
     O << "  error opening file for writing!\n";
   }