Add supportsTypedPointers and SimpleLoopUnswitch(#281)

Co-authored-by: Gabriel Baraldi <[email protected]>

Author: vchuravy

Manifest.toml

@@ -34,9 +34,9 @@ version = "1.4.1"
 
 [[LLVMExtra_jll]]
 deps = ["Artifacts", "JLLWrappers", "LazyArtifacts", "Libdl", "TOML"]
-git-tree-sha1 = "1fbc1c5dac58bd6c6183347b67ab6c2f0ce6c582"
+git-tree-sha1 = "070e4b5b65827f82c16ae0916376cb47377aa1b5"
 uuid = "dad2f222-ce93-54a1-a47d-0025e8a3acab"
-version = "0.0.17+0"
+version = "0.0.18+0"
 
 [[LazyArtifacts]]
 deps = ["Artifacts", "Pkg"]

Project.toml

@@ -11,5 +11,5 @@ Unicode = "4ec0a83e-493e-50e2-b9ac-8f72acf5a8f5"
 
 [compat]
 CEnum = "0.2, 0.3, 0.4"
-LLVMExtra_jll = "=0.0.17"
+LLVMExtra_jll = "=0.0.18"
 julia = "1.6"

deps/LLVMExtra/include/LLVMExtra.h

@@ -30,6 +30,7 @@ void LLVMAddVectorCombinePass(LLVMPassManagerRef PM);
 void LLVMAddSpeculativeExecutionIfHasBranchDivergencePass(LLVMPassManagerRef PM);
 void LLVMAddSimpleLoopUnrollPass(LLVMPassManagerRef PM);
 void LLVMAddInductiveRangeCheckEliminationPass(LLVMPassManagerRef PM);
+void LLVMAddSimpleLoopUnswitchLegacyPass(LLVMPassManagerRef PM);
 
 #if LLVM_VERSION_MAJOR < 12
 void LLVMAddInstructionSimplifyPass(LLVMPassManagerRef PM);
@@ -157,5 +158,9 @@ LLVMValueRef LLVMMetadataAsValue2(LLVMContextRef C, LLVMMetadataRef Metadata);
 void LLVMReplaceAllMetadataUsesWith(LLVMValueRef Old, LLVMValueRef New);
 void LLVMReplaceMDNodeOperandWith(LLVMMetadataRef MD, unsigned I, LLVMMetadataRef New);
 
+#if LLVM_VERSION_MAJOR >= 12
+LLVMBool LLVMContextSupportsTypedPointers(LLVMContextRef C);
+#endif
+
 LLVM_C_EXTERN_C_END
 #endif

deps/LLVMExtra/lib/llvm-api.cpp

@@ -15,6 +15,7 @@
 #include <llvm/Support/TargetSelect.h>
 #include <llvm/Transforms/IPO.h>
 #include <llvm/Transforms/Scalar.h>
+#include <llvm/Transforms/Scalar/SimpleLoopUnswitch.h>
 #include <llvm/Transforms/Vectorize.h>
 #if LLVM_VERSION_MAJOR < 12
 #include <llvm/Transforms/Scalar/InstSimplifyPass.h>
@@ -101,6 +102,11 @@ void LLVMAddInductiveRangeCheckEliminationPass(LLVMPassManagerRef PM)
     unwrap(PM)->add(createInductiveRangeCheckEliminationPass());
 }
 
+void LLVMAddSimpleLoopUnswitchLegacyPass(LLVMPassManagerRef PM)
+{
+    unwrap(PM)->add(createSimpleLoopUnswitchLegacyPass());
+}
+
 #if LLVM_VERSION_MAJOR < 12
 void LLVMAddInstructionSimplifyPass(LLVMPassManagerRef PM)
 {
@@ -544,3 +550,9 @@ void LLVMReplaceAllMetadataUsesWith(LLVMValueRef Old, LLVMValueRef New) {
 void LLVMReplaceMDNodeOperandWith(LLVMMetadataRef MD, unsigned I, LLVMMetadataRef New) {
     unwrap<MDNode>(MD)->replaceOperandWith(I, unwrap(New));
 }
+
+#if LLVM_VERSION_MAJOR > 12
+LLVMBool LLVMContextSupportsTypedPointers(LLVMContextRef C) {
+  return unwrap(C)->supportsTypedPointers();
+}
+#endif

lib/libLLVM_extra.jl

@@ -63,6 +63,10 @@ function LLVMAddInductiveRangeCheckEliminationPass(PM)
     ccall((:LLVMAddInductiveRangeCheckEliminationPass, libLLVMExtra), Cvoid, (LLVMPassManagerRef,), PM)
 end
 
+function LLVMAddSimpleLoopUnswitchLegacyPass(PM)
+    ccall((:LLVMAddSimpleLoopUnswitchLegacyPass, libLLVMExtra), Cvoid, (LLVMPassManagerRef,), PM)
+end
+
 if version() < v"12"
 function LLVMAddInstructionSimplifyPass(PM)
     ccall((:LLVMAddInstructionSimplifyPass, libLLVMExtra), Cvoid, (LLVMPassManagerRef,), PM)
@@ -406,3 +410,9 @@ end
 function LLVMReplaceMDNodeOperandWith(MD, I, New)
     ccall((:LLVMReplaceMDNodeOperandWith, libLLVMExtra), Cvoid, (LLVMMetadataRef, Cuint, LLVMMetadataRef), MD, I, New)
 end
+
+if version() > v"12"
+function LLVMContextSupportsTypedPointers(Ctx)
+    ccall((:LLVMContextSupportsTypedPointers, libLLVMExtra), LLVMBool, (LLVMContextRef,), Ctx)
+end
+end

src/core/context.jl

@@ -27,6 +27,11 @@ end
 
 GlobalContext() = Context(API.LLVMGetGlobalContext())
 
+if version() > v"13.0.0"
+    supports_typed_pointers(ctx::Context) = API.LLVMContextSupportsTypedPointers(ctx) == 1
+else
+    supports_typed_pointers(ctx::Context) = false
+end
 
 ## wrapper exception type
 

test/core.jl

@@ -25,6 +25,10 @@ Context() do ctx end
 
 @dispose ctx=Context() begin end
 
+@dispose ctx=Context() begin
+    @test LLVM.supports_typed_pointers(ctx) isa Bool
+end
+
 end