1 files changed, 9 insertions, 1 deletions
diff --git a/Driver/ASTConsumers.cpp b/Driver/ASTConsumers.cpp
index 4a2ef778a0..959d4baeb0 100644
--- a/Driver/ASTConsumers.cpp
+++ b/Driver/ASTConsumers.cpp
@@ -652,15 +652,23 @@ public:
   virtual void Initialize(ASTContext &Context) {
     Gen->Initialize(Context);
   }
+
+  virtual void InitializeTU(TranslationUnit& TU) {
+    Gen->InitializeTU(TU);
+  }
   
   virtual void HandleTopLevelDecl(Decl *D) {
     Gen->HandleTopLevelDecl(D);
   }
   
+  virtual void HandleTranslationUnit(TranslationUnit& TU) {
+    Gen->HandleTranslationUnit(TU);
+  }
+  
   virtual void HandleTagDeclDefinition(TagDecl *D) {
     Gen->HandleTagDeclDefinition(D);
   }
-  
+
   virtual ~LLVMCodeGenWriter() {
     llvm::OwningPtr<llvm::Module> CodeGenModule(Gen->ReleaseModule());