Add GetModule accessor to ModuleBuilder

llvm-svn: 57924

2 files changed, 7 insertions, 2 deletions

diff --git a/clang/include/clang/CodeGen/ModuleBuilder.h b/clang/include/clang/CodeGen/ModuleBuilder.h
index 0f9b39341bd..4642e4eeeac 100644
--- a/clang/include/clang/CodeGen/ModuleBuilder.h
+++ b/clang/include/clang/CodeGen/ModuleBuilder.h
@@ -26,8 +26,9 @@ namespace clang {
   struct LangOptions;
   
   class CodeGenerator : public ASTConsumer {
-    public:
-      virtual llvm::Module* ReleaseModule() = 0;    
+  public:
+    virtual llvm::Module* GetModule() = 0;
+    virtual llvm::Module* ReleaseModule() = 0;    
   };
   
   CodeGenerator *CreateLLVMCodeGen(Diagnostic &Diags,
diff --git a/clang/lib/CodeGen/ModuleBuilder.cpp b/clang/lib/CodeGen/ModuleBuilder.cpp
index dafc0f5e258..be6e171bc45 100644
--- a/clang/lib/CodeGen/ModuleBuilder.cpp
+++ b/clang/lib/CodeGen/ModuleBuilder.cpp
@@ -48,6 +48,10 @@ namespace {
     
     virtual ~CodeGeneratorImpl() {}
     
+    virtual llvm::Module* GetModule() {
+      return M.get();
+    }
+    
     virtual llvm::Module* ReleaseModule() {
       return M.take();
     }