Add a couple of missing passes.

Author: maleadt

deps/LLVMExtra/include/LLVMExtra.h

@@ -33,6 +33,9 @@ void LLVMAddLoopFusePass(LLVMPassManagerRef PM);
 void LLVMAddLoopLoadEliminationPass(LLVMPassManagerRef PM);
 void LLVMAddLoadStoreVectorizerPass(LLVMPassManagerRef PM);
 void LLVMAddVectorCombinePass(LLVMPassManagerRef PM);
+void LLVMAddSpeculativeExecutionIfHasBranchDivergencePass(LLVMPassManagerRef PM);
+void LLVMAddSimpleLoopUnrollPass(LLVMPassManagerRef PM);
+void LLVMAddInductiveRangeCheckEliminationPass(LLVMPassManagerRef PM);
 
 #if LLVM_VERSION_MAJOR < 12
 void LLVMAddInstructionSimplifyPass(LLVMPassManagerRef PM);

deps/LLVMExtra/lib/llvm-api.cpp

@@ -116,6 +116,21 @@ void LLVMAddVectorCombinePass(LLVMPassManagerRef PM)
     unwrap(PM)->add(createVectorCombinePass());
 }
 
+void LLVMAddSpeculativeExecutionIfHasBranchDivergencePass(LLVMPassManagerRef PM)
+{
+    unwrap(PM)->add(createSpeculativeExecutionIfHasBranchDivergencePass());
+}
+
+void LLVMAddSimpleLoopUnrollPass(LLVMPassManagerRef PM)
+{
+    unwrap(PM)->add(createSimpleLoopUnrollPass());
+}
+
+void LLVMAddInductiveRangeCheckEliminationPass(LLVMPassManagerRef PM)
+{
+    unwrap(PM)->add(createInductiveRangeCheckEliminationPass());
+}
+
 #if LLVM_VERSION_MAJOR < 12
 void LLVMAddInstructionSimplifyPass(LLVMPassManagerRef PM)
 {

lib/libLLVM_extra.jl

@@ -75,6 +75,18 @@ function LLVMAddVectorCombinePass(PM)
     ccall((:LLVMAddVectorCombinePass, libLLVMExtra), Cvoid, (LLVMPassManagerRef,), PM)
 end
 
+function LLVMAddSpeculativeExecutionIfHasBranchDivergencePass(PM)
+    ccall((:LLVMAddSpeculativeExecutionIfHasBranchDivergencePass, libLLVMExtra), Cvoid, (LLVMPassManagerRef,), PM)
+end
+
+function LLVMAddSimpleLoopUnrollPass(PM)
+    ccall((:LLVMAddSimpleLoopUnrollPass, libLLVMExtra), Cvoid, (LLVMPassManagerRef,), PM)
+end
+
+function LLVMAddInductiveRangeCheckEliminationPass(PM)
+    ccall((:LLVMAddInductiveRangeCheckEliminationPass, libLLVMExtra), Cvoid, (LLVMPassManagerRef,), PM)
+end
+
 if version() < v"12"
 function LLVMAddInstructionSimplifyPass(PM)
     ccall((:LLVMAddInstructionSimplifyPass, libLLVMExtra), Cvoid, (LLVMPassManagerRef,), PM)

src/transform.jl

@@ -103,7 +103,8 @@ define_transforms([
     :SimplifyLibCalls, :TailCallElimination, :ConstantPropagation, :DemoteMemoryToRegister,
     :Verifier, :CorrelatedValuePropagation, :EarlyCSE, :EarlyCSEMemSSA,
     :LowerExpectIntrinsic, :TypeBasedAliasAnalysis, :ScopedNoAliasAA, :BasicAliasAnalysis,
-    :MergeFunctions
+    :MergeFunctions, :SpeculativeExecutionIfHasBranchDivergence, :SimpleLoopUnroll,
+    :InductiveRangeCheckElimination
 ])
 
 export scalar_repl_aggregates!, scalar_repl_aggregates_ssa!, cfgsimplification!

test/transform.jl

@@ -84,6 +84,9 @@ ModulePassManager() do pm
     loop_vectorize!(pm)
     slpvectorize!(pm)
     load_store_vectorizer!(pm)
+    speculative_execution_if_has_branch_divergence!(pm)
+    simple_loop_unroll!(pm)
+    inductive_range_check_elimination!(pm)
 
     argument_promotion!(pm)
     constant_merge!(pm)