Let LLVMCodeGenWriter forward all the methods defined in ASTConsumer, to

prevent unexpected surprises later on.

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

diff --git a/Driver/ASTConsumers.cpp b/Driver/ASTConsumers.cpp
index 4a2ef778a09db1aa171409e8b66d35fa3b096265..959d4baeb07f8f137b857684df26ece8ecf13d1a 100644 (file)
--- 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());