2 files changed, 7 insertions, 2 deletions
diff --git a/include/clang/CodeGen/ModuleBuilder.h b/include/clang/CodeGen/ModuleBuilder.h
index 0f9b39341b..4642e4eeea 100644
--- a/include/clang/CodeGen/ModuleBuilder.h
+++ b/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/lib/CodeGen/ModuleBuilder.cpp b/lib/CodeGen/ModuleBuilder.cpp
index dafc0f5e25..be6e171bc4 100644
--- a/lib/CodeGen/ModuleBuilder.cpp
+++ b/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();
     }