Merge pull request #275 from maleadt/vc/cpu_feat

vchuravy · web-flow · commit 6a48caea6b81 · 2021-12-10T17:11:27.000-05:00
add Julia wrapper for CPU features pass
diff --git a/lib/libLLVM_julia.jl b/lib/libLLVM_julia.jl
@@ -55,3 +55,7 @@ end
 function LLVMAddJuliaLICMPass(PM)
     ccall(:LLVMExtraJuliaLICMPass,Cvoid,(LLVMPassManagerRef,), PM)
 end
+
+function LLVMAddCPUFeaturesPass(PM)
+    ccall(:LLVMExtraAddCPUFeaturesPass,Cvoid,(LLVMPassManagerRef,), PM)
+end
diff --git a/src/interop/passes.jl b/src/interop/passes.jl
@@ -1,7 +1,7 @@
 export alloc_opt!, barrier_noop!, gc_invariant_verifier!, lower_exc_handlers!,
        combine_mul_add!, multi_versioning!, propagate_julia_addrsp!, lower_ptls!,
        lower_simdloop!, late_lower_gc_frame!, final_lower_gc!, remove_julia_addrspaces!,
-       demote_float16!, remove_ni!, julia_licm!
+       demote_float16!, remove_ni!, julia_licm!, cpu_features!
 
 alloc_opt!(pm::PassManager) = API.LLVMAddAllocOptPass(pm)
 
@@ -34,3 +34,9 @@ demote_float16!(pm::PassManager) = API.LLVMAddDemoteFloat16Pass(pm)
 remove_ni!(pm::PassManager) = API.LLVMAddRemoveNIPass(pm)
 
 julia_licm!(pm::PassManager) = API.LLVMAddJuliaLICMPass(pm)
+
+if VERSION >= v"1.8.0-DEV.1120"
+    cpu_features!(pm::PassManager) = API.LLVMAddCPUFeaturesPass(pm)
+else
+    cpu_features!(pm::PassManager) = nothing
+end
diff --git a/test/interop.jl b/test/interop.jl
@@ -122,6 +122,7 @@ lower_simdloop!(pm)
 remove_ni!(pm)
 late_lower_gc_frame!(pm)
 final_lower_gc!(pm)
+cpu_features!(pm)
 
 end
 end
