diff --git a/clang/lib/Interpreter/IncrementalAction.cpp b/clang/lib/Interpreter/IncrementalAction.cpp index 3d489fce54bc6..e2b9d13017ada 100644 --- a/clang/lib/Interpreter/IncrementalAction.cpp +++ b/clang/lib/Interpreter/IncrementalAction.cpp @@ -120,6 +120,17 @@ std::unique_ptr IncrementalAction::GenModule() { return nullptr; } +void IncrementalAction::discardCurrentCodeGenModule() { + if (CodeGenerator *CG = getCodeGen()) { + if (auto *CurM = CG->GetModule()) { + llvm::LLVMContext &Ctx = CurM->getContext(); + std::string Name = CurM->getName().str(); + std::unique_ptr Dead(CG->ReleaseModule()); + CG->StartModule(Name, Ctx); + } + } +} + CodeGenerator *IncrementalAction::getCodeGen() const { FrontendAction *WrappedAct = getWrapped(); if (!WrappedAct || !WrappedAct->hasIRSupport()) diff --git a/clang/lib/Interpreter/IncrementalAction.h b/clang/lib/Interpreter/IncrementalAction.h index 725cdd0c27cf4..485cfaa45f793 100644 --- a/clang/lib/Interpreter/IncrementalAction.h +++ b/clang/lib/Interpreter/IncrementalAction.h @@ -74,6 +74,8 @@ class IncrementalAction : public WrapperFrontendAction { /// Generate an LLVM module for the most recent parsed input. std::unique_ptr GenModule(); + + void discardCurrentCodeGenModule(); }; class InProcessPrintingASTConsumer final : public MultiplexConsumer { diff --git a/clang/lib/Interpreter/IncrementalParser.cpp b/clang/lib/Interpreter/IncrementalParser.cpp index bf08911e23533..53379603c26da 100644 --- a/clang/lib/Interpreter/IncrementalParser.cpp +++ b/clang/lib/Interpreter/IncrementalParser.cpp @@ -82,6 +82,7 @@ IncrementalParser::ParseOrWrapTopLevelDecl() { DiagnosticsEngine &Diags = S.getDiagnostics(); if (Diags.hasErrorOccurred()) { + Act->discardCurrentCodeGenModule(); CleanUpPTU(C.getTranslationUnitDecl()); Diags.Reset(/*soft=*/true);