Make CIRGenerator::mlirContext a shared pointer

Author: andykaylor

clang/include/clang/CIR/CIRGenerator.h

@@ -44,7 +44,7 @@ class CIRGenerator : public clang::ASTConsumer {
   const clang::CodeGenOptions &codeGenOpts;
 
 protected:
-  std::unique_ptr<mlir::MLIRContext> mlirContext;
+  std::shared_ptr<mlir::MLIRContext> mlirContext;
   std::unique_ptr<clang::CIRGen::CIRGenModule> cgm;
 
 public:
@@ -55,8 +55,8 @@ class CIRGenerator : public clang::ASTConsumer {
   void Initialize(clang::ASTContext &astContext) override;
   bool HandleTopLevelDecl(clang::DeclGroupRef group) override;
   mlir::ModuleOp getModule() const;
-  std::unique_ptr<mlir::MLIRContext> takeContext() {
-    return std::move(mlirContext);
+  std::shared_ptr<mlir::MLIRContext> getContext() {
+    return mlirContext;
   };
 };
 

clang/include/clang/CIR/LowerToLLVM.h

@@ -33,4 +33,4 @@ lowerDirectlyFromCIRToLLVMIR(mlir::ModuleOp M, llvm::LLVMContext &Ctx);
 } // namespace direct
 } // namespace cir
 
-#endif // CLANG_CIR_LOWERTOLLVM_H
+#endif // CLANG_CIR_LOWERTOLLVM_H

clang/lib/CIR/CodeGen/CIRGenerator.cpp

@@ -34,7 +34,7 @@ void CIRGenerator::Initialize(ASTContext &astContext) {
 
   this->astContext = &astContext;
 
-  mlirContext = std::make_unique<mlir::MLIRContext>();
+  mlirContext = std::make_shared<mlir::MLIRContext>();
   mlirContext->loadDialect<cir::CIRDialect>();
   cgm = std::make_unique<clang::CIRGen::CIRGenModule>(
       *mlirContext.get(), astContext, codeGenOpts, diags);

clang/lib/CIR/FrontendAction/CIRGenAction.cpp

@@ -32,7 +32,7 @@ getBackendActionFromOutputType(CIRGenAction::OutputType Action) {
 
 static std::unique_ptr<llvm::Module> lowerFromCIRToLLVMIR(
     const clang::FrontendOptions &FEOpts, mlir::ModuleOp MLIRModule,
-    std::unique_ptr<mlir::MLIRContext> MLIRCtx, llvm::LLVMContext &LLVMCtx) {
+    std::shared_ptr<mlir::MLIRContext> MLIRCtx, llvm::LLVMContext &LLVMCtx) {
   return direct::lowerDirectlyFromCIRToLLVMIR(MLIRModule, LLVMCtx);
 }
 
@@ -72,7 +72,7 @@ class CIRGenConsumer : public clang::ASTConsumer {
   void HandleTranslationUnit(ASTContext &C) override {
     Gen->HandleTranslationUnit(C);
     mlir::ModuleOp MlirModule = Gen->getModule();
-    auto MLIRCtx = Gen->takeContext();
+    std::shared_ptr<mlir::MLIRContext> MLIRCtx = Gen->getContext();
     switch (Action) {
     case CIRGenAction::OutputType::EmitCIR:
       if (OutputStream && MlirModule) {
@@ -84,7 +84,7 @@ class CIRGenConsumer : public clang::ASTConsumer {
     case CIRGenAction::OutputType::EmitLLVM: {
       llvm::LLVMContext LLVMCtx;
       auto LLVMModule = lowerFromCIRToLLVMIR(CI.getFrontendOpts(), MlirModule,
-                                             std::move(MLIRCtx), LLVMCtx);
+                                             MLIRCtx, LLVMCtx);
 
       BackendAction BEAction = getBackendActionFromOutputType(Action);
       emitBackendOutput(
@@ -135,4 +135,4 @@ EmitCIRAction::EmitCIRAction(mlir::MLIRContext *MLIRCtx)
 
 void EmitLLVMAction::anchor() {}
 EmitLLVMAction::EmitLLVMAction(mlir::MLIRContext *MLIRCtx)
-    : CIRGenAction(OutputType::EmitLLVM, MLIRCtx) {}
+    : CIRGenAction(OutputType::EmitLLVM, MLIRCtx) {}

clang/lib/CIR/Lowering/CMakeLists.txt

@@ -1 +1 @@
-add_subdirectory(DirectToLLVM)
+add_subdirectory(DirectToLLVM)