[CIR] Add runAtStartOfConvertCIRToMLIRPass() to ConvertCIRToMLIRPass

This allows injecting some dialects and patterns to the
cir::ConvertCIRToMLIRPass lowering pass.

Author: keryell

clang/include/clang/CIR/LowerToMLIR.h

@@ -13,6 +13,7 @@
 #define CLANG_CIR_LOWERTOMLIR_H
 
 #include "mlir/Transforms/DialectConversion.h"
+#include <functional>
 
 namespace cir {
 
@@ -21,6 +22,8 @@ void populateCIRLoopToSCFConversionPatterns(mlir::RewritePatternSet &patterns,
 
 mlir::TypeConverter prepareTypeConverter();
 
+void runAtStartOfConvertCIRToMLIRPass(std::function<void(mlir::ConversionTarget)>);
+
 mlir::ModuleOp
 lowerFromCIRToMLIRToLLVMDialect(mlir::ModuleOp theModule,
                                 mlir::MLIRContext *mlirCtx = nullptr);

clang/lib/CIR/Lowering/ThroughMLIR/LowerCIRToMLIR.cpp

@@ -10,6 +10,8 @@
 //
 //===----------------------------------------------------------------------===//
 
+#include <functional>
+
 #include "LowerToMLIRHelpers.h"
 #include "mlir/Conversion/AffineToStandard/AffineToStandard.h"
 #include "mlir/Conversion/ArithToLLVM/ArithToLLVM.h"
@@ -92,6 +94,8 @@ struct ConvertCIRToMLIRPass
   }
 
   StringRef getArgument() const override { return "cir-to-mlir"; }
+
+  inline static std::function<void(mlir::ConversionTarget)> runAtStartHook;
 };
 
 class CIRCallOpLowering : public mlir::OpConversionPattern<cir::CallOp> {
@@ -1460,10 +1464,20 @@ void ConvertCIRToMLIRPass::runOnOperation() {
                        mlir::math::MathDialect, mlir::vector::VectorDialect>();
   target.addIllegalDialect<cir::CIRDialect>();
 
+  if (runAtStartHook)
+    runAtStartHook(target);
+
   if (failed(applyPartialConversion(module, target, std::move(patterns))))
     signalPassFailure();
 }
 
+/// Set a hook to be called just before applying the dialect conversion so other
+/// dialects or patterns can be added
+void runAtStartOfConvertCIRToMLIRPass(
+    std::function<void(mlir::ConversionTarget)> hook) {
+  ConvertCIRToMLIRPass::runAtStartHook = std::move(hook);
+}
+
 mlir::ModuleOp lowerFromCIRToMLIRToLLVMDialect(mlir::ModuleOp theModule,
                                                mlir::MLIRContext *mlirCtx) {
   llvm::TimeTraceScope scope("Lower from CIR to MLIR To LLVM Dialect");