add PassManagerBuilder.addExtension

Author: vchuravy

deps/LLVMExtra/include/LLVMExtra.h

@@ -165,5 +165,27 @@ LLVMValueRef LLVMBuildCallWithOpBundle(LLVMBuilderRef B, LLVMValueRef Fn,
                                        const char *Name);
 void LLVMAdjustPassManager(LLVMTargetMachineRef TM, LLVMPassManagerBuilderRef PMB);
 
+typedef void (*LLVMPassManagerBuilderExtensionFunction)(
+    void *Ctx, LLVMPassManagerBuilderRef PMB, LLVMPassManagerRef PM);
+
+typedef enum {
+    EP_EarlyAsPossible,
+    EP_ModuleOptimizerEarly,
+    EP_LoopOptimizerEnd,
+    EP_ScalarOptimizerLate,
+    EP_OptimizerLast,
+    EP_VectorizerStart,
+    EP_EnabledOnOptLevel0,
+    EP_Peephole,
+    EP_LateLoopOptimizations,
+    EP_CGSCCOptimizerLate,
+    EP_FullLinkTimeOptimizationEarly,
+    EP_FullLinkTimeOptimizationLast,
+} LLVMPassManagerBuilderExtensionPointTy;
+
+void LLVMPassManagerBuilderAddExtension(LLVMPassManagerBuilderRef PMB,
+                                        LLVMPassManagerBuilderExtensionPointTy Ty,
+                                        LLVMPassManagerBuilderExtensionFunction Fn, void *Ctx);
+
 LLVM_C_EXTERN_C_END
 #endif

deps/LLVMExtra/lib/llvm-api.cpp

@@ -22,6 +22,7 @@
 #include <llvm/Transforms/Utils/Cloning.h>
 #include <llvm/Transforms/Utils/ModuleUtils.h>
 #include <llvm/Target/TargetMachine.h>
+#include <llvm/Transforms/IPO/PassManagerBuilder.h>
 
 using namespace llvm;
 using namespace llvm::legacy;
@@ -551,7 +552,15 @@ LLVMValueRef LLVMBuildCallWithOpBundle(LLVMBuilderRef B, LLVMValueRef Fn,
 }
 
 DEFINE_SIMPLE_CONVERSION_FUNCTIONS(TargetMachine, LLVMTargetMachineRef)
-DEFINE_SIMPLE_CONVERSION_FUNCTIONS(PassManagerBuilder, LLVMPassManagerBuilderRef)
 void LLVMAdjustPassManager(LLVMTargetMachineRef TM, LLVMPassManagerBuilderRef PMB) {
   unwrap(TM)->adjustPassManager(*unwrap(PMB));
 }
+
+void LLVMPassManagerBuilderAddExtension(LLVMPassManagerBuilderRef PMB,
+                                        LLVMPassManagerBuilderExtensionPointTy Ty,
+                                        LLVMPassManagerBuilderExtensionFunction Fn, void *Ctx) {
+  PassManagerBuilder &Builder = *unwrap(PMB);
+  Builder.addExtension((PassManagerBuilder::ExtensionPointTy)Ty,
+    [&](const PassManagerBuilder &Builder, legacy::PassManagerBase &PM) {
+        Fn(Ctx, wrap(const_cast<PassManagerBuilder*>(&Builder)), wrap(&PM)); });
+}