MC CFG: Keep pointer to parent MCModule in created MCFunctions.

[oota-llvm.git] / lib / MC / MCFunction.cpp

diff --git a/lib/MC/MCFunction.cpp b/lib/MC/MCFunction.cpp
index 473d07bf9949de802599d4764463c8f656c68179..cb2504668a328a20674deaa84d52ee5d652eaf4a 100644 (file)
--- a/lib/MC/MCFunction.cpp
+++ b/lib/MC/MCFunction.cpp
@@ -9,15 +9,15 @@
 
 #include "llvm/MC/MCFunction.h"
 #include "llvm/MC/MCAtom.h"
-#include "llvm/Support/raw_ostream.h"
+#include "llvm/MC/MCModule.h"
 #include <algorithm>
 
 using namespace llvm;
 
 // MCFunction
 
-MCFunction::MCFunction(StringRef Name)
-  : Name(Name)
+MCFunction::MCFunction(StringRef Name, MCModule *Parent)
+  : Name(Name), ParentModule(Parent)
 {}
 
 MCFunction::~MCFunction() {