Allow getting the module from a call graph

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

diff --git a/include/llvm/Analysis/CallGraph.h b/include/llvm/Analysis/CallGraph.h
index 305d43dd758ab514a2ddb170090938eeb2e29541..0e9f614e382d7a37124646ad23d20228be4daa24 100644 (file)
--- a/include/llvm/Analysis/CallGraph.h
+++ b/include/llvm/Analysis/CallGraph.h
@@ -100,6 +100,10 @@ public:
         CallGraphNode *getRoot()       { return Root; }
   const CallGraphNode *getRoot() const { return Root; }
 
+  /// getModule - Return the module the call graph corresponds to.
+  ///
+  Module &getModule() const { return *Mod; }
+
   inline       iterator begin()       { return FunctionMap.begin(); }
   inline       iterator end()         { return FunctionMap.end();   }
   inline const_iterator begin() const { return FunctionMap.begin(); }