Merge pull request #215 from maleadt/vc/passes

Add passes exported from Julia

Author: maleadt

lib/libLLVM_extra.jl

@@ -186,6 +186,42 @@ function LLVMAddJuliaLICMPass(PM)
 end
 end
 
+if VERSION >= v"1.6.0-DEV.1503"
+function LLVMAddDivRemPairsPass(PM)
+    ccall(:LLVMExtraAddDivRemPairsPass,Cvoid,(LLVMPassManagerRef,), PM)
+end
+end
+
+if VERSION >= v"1.6.0-DEV.1503"
+function LLVMAddLoopDistributePass(PM)
+    ccall(:LLVMExtraAddLoopDistributePass,Cvoid,(LLVMPassManagerRef,), PM)
+end
+end
+
+if VERSION >= v"1.6.0-DEV.1503"
+function LLVMAddLoopFusePass(PM)
+    ccall(:LLVMExtraAddLoopFusePass,Cvoid,(LLVMPassManagerRef,), PM)
+end
+end
+
+if VERSION >= v"1.6.0-DEV.1503"
+function LLVMAddLoopLoadEliminationPass(PM)
+    ccall(:LLVMExtraLoopLoadEliminationPass,Cvoid,(LLVMPassManagerRef,), PM)
+end
+end
+
+if VERSION >= v"1.6.0-DEV.1503"
+function LLVMAddLoadStoreVectorizerPass(PM)
+    ccall(:LLVMExtraAddLoadStoreVectorizerPass,Cvoid,(LLVMPassManagerRef,), PM)
+end
+end
+
+if VERSION >= v"1.6.0-DEV.1503"
+function LLVMAddInstSimplifyPass(PM)
+    ccall(:LLVMExtraAddInstructionSimplifyPass,Cvoid,(LLVMPassManagerRef,), PM)
+end
+end
+
 function LLVMGetValueContext(V)
     ccall(:LLVMExtraGetValueContext,LLVMContextRef,(LLVMValueRef,),V)
 end

src/transform.jl

@@ -53,7 +53,7 @@ populate!(mpm::ModulePassManager, pmb::PassManagerBuilder) =
 
 ## auxiliary
 
-function define_transforms(transforms)
+function define_transforms(transforms, available=true)
     for transform in transforms
         api_fname = Symbol(:LLVM, :Add, transform, :Pass)
 
@@ -78,9 +78,16 @@ function define_transforms(transforms)
         end
         jl_fname = Symbol(join(lowercase.(groups), '_'), '!')
 
-        @eval begin
-            export $jl_fname
-            $jl_fname(pm::PassManager) = API.$api_fname(pm)
+        if available
+            @eval begin
+                export $jl_fname
+                $jl_fname(pm::PassManager) = API.$api_fname(pm)
+            end
+        else
+            @eval begin
+                export $jl_fname
+                $jl_fname(pm::PassManager) = nothing
+            end
         end
     end
 
@@ -101,27 +108,27 @@ define_transforms([
     :LowerExpectIntrinsic, :TypeBasedAliasAnalysis, :ScopedNoAliasAA, :BasicAliasAnalysis
 ])
 
-export scalar_repl_aggregates!, scalar_repl_aggregates_ssa!, dce!
+export scalar_repl_aggregates!, scalar_repl_aggregates_ssa!
 
 scalar_repl_aggregates!(pm::PassManager, threshold::Integer) =
     API.LLVMAddScalarReplAggregatesPassWithThreshold(pm, Cint(threshold))
 
 scalar_repl_aggregates_ssa!(pm::PassManager) =
     API.LLVMAddScalarReplAggregatesPassSSA(pm)
 
-if version() >= v"10.0"
-dce!(pm::PassManager) =
-    API.LLVMAddDCEPass(pm)
-else
-dce!(pm::PassManager) = nothing
-end
+define_transforms([:DCE], version() >= v"10.0")
+
+define_transforms([
+    :DivRemPairs, :LoopDistribute, :LoopFuse, :LoopLoadElimination, :InstSimplify
+], VERSION >= v"1.6.0-DEV.1503")
 
 ## vectorization transformations
 
 define_transforms([
     :LoopVectorize, :SLPVectorize
 ])
 
+define_transforms([:LoadStoreVectorizer], VERSION >= v"1.6.0-DEV.1503")
 
 ## interprocedural transformations
 

test/transform.jl

@@ -38,8 +38,10 @@ ModulePassManager() do pm
     scalarizer!(pm)
     merged_load_store_motion!(pm)
     gvn!(pm)
+    div_rem_pairs!(pm)
     ind_var_simplify!(pm)
     instruction_combining!(pm)
+    inst_simplify!(pm)
     jump_threading!(pm)
     licm!(pm)
     loop_deletion!(pm)
@@ -48,7 +50,11 @@ ModulePassManager() do pm
     loop_reroll!(pm)
     loop_unroll!(pm)
     loop_unswitch!(pm)
+    loop_distribute!(pm)
+    loop_fuse!(pm)
+    loop_load_elimination!(pm)
     mem_cpy_opt!(pm)
+
     partially_inline_lib_calls!(pm)
     lower_switch!(pm)
     promote_memory_to_register!(pm)
@@ -71,6 +77,7 @@ ModulePassManager() do pm
 
     loop_vectorize!(pm)
     slpvectorize!(pm)
+    load_store_vectorizer!(pm)
 
     argument_promotion!(pm)
     constant_merge!(pm)