[CIR] Add runAtStartOfConvertCIRToLLVMPass() to ConvertCIRToLLVMPass

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

Author: keryell

clang/include/clang/CIR/LowerToLLVM.h

@@ -13,7 +13,9 @@
 #define CLANG_CIR_LOWERTOLLVM_H
 
 #include "mlir/Pass/Pass.h"
+#include "mlir/Transforms/DialectConversion.h"
 
+#include <functional>
 #include <memory>
 
 namespace llvm {
@@ -33,7 +35,9 @@ std::unique_ptr<llvm::Module> lowerDirectlyFromCIRToLLVMIR(
     mlir::ModuleOp theModule, llvm::LLVMContext &llvmCtx,
     bool disableVerifier = false, bool disableCCLowering = false,
     bool disableDebugInfo = false);
-}
+void runAtStartOfConvertCIRToLLVMPass(
+    std::function<void(mlir::ConversionTarget)>);
+} // namespace direct
 
 // Lower directly from pristine CIR to LLVMIR.
 std::unique_ptr<llvm::Module>

clang/lib/CIR/Lowering/DirectToLLVM/LowerToLLVM.cpp

@@ -1352,6 +1352,8 @@ struct ConvertCIRToLLVMPass
 
   void processCIRAttrs(mlir::ModuleOp moduleOp);
 
+  inline static std::function<void(mlir::ConversionTarget)> runAtStartHook;
+
   StringRef getDescription() const override {
     return "Convert the prepared CIR dialect module to LLVM dialect";
   }
@@ -4692,6 +4694,9 @@ void ConvertCIRToLLVMPass::runOnOperation() {
                     // ,YieldOp
                     >();
   // clang-format on
+  if (runAtStartHook)
+    runAtStartHook(target);
+
   target.addLegalDialect<mlir::LLVM::LLVMDialect>();
   target.addIllegalDialect<mlir::BuiltinDialect, cir::CIRDialect,
                            mlir::func::FuncDialect>();
@@ -4729,6 +4734,13 @@ void ConvertCIRToLLVMPass::runOnOperation() {
   buildGlobalAnnotationsVar(stringGlobalsMap, argStringGlobalsMap, argsVarMap);
 }
 
+/// Set a hook to be called just before applying the dialect conversion so other
+/// dialects or patterns can be added
+void runAtStartOfConvertCIRToLLVMPass(
+    std::function<void(mlir::ConversionTarget)> hook) {
+  ConvertCIRToLLVMPass::runAtStartHook = std::move(hook);
+}
+
 std::unique_ptr<mlir::Pass> createConvertCIRToLLVMPass() {
   return std::make_unique<ConvertCIRToLLVMPass>();
 }