Merge pull request #258 from maleadt/tb/delete

Add a wrapper for llvm::Function's deleteBody.

Author: maleadt

Manifest.toml

@@ -34,9 +34,9 @@ version = "1.3.0"
 
 [[LLVMExtra_jll]]
 deps = ["Artifacts", "JLLWrappers", "Libdl", "Pkg"]
-git-tree-sha1 = "2d5a0044d6505f4771b5c82de87393f0c9741154"
+git-tree-sha1 = "873e7962f14f6bdd8a0e10552d964ec0a7c69f3b"
 uuid = "dad2f222-ce93-54a1-a47d-0025e8a3acab"
-version = "0.0.8+0"
+version = "0.0.9+0"
 
 [[LibCURL]]
 deps = ["LibCURL_jll", "MozillaCACerts_jll"]

Project.toml

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

deps/LLVMExtra/include/LLVMExtra.h

@@ -131,5 +131,7 @@ void LLVMCloneFunctionInto(LLVMValueRef NewFunc, LLVMValueRef OldFunc,
                            LLVMValueRef (*Materializer)(LLVMValueRef, void *),
                            void *MaterializerData);
 
+void LLVMFunctionDeleteBody(LLVMValueRef Func);
+
 LLVM_C_EXTERN_C_END
 #endif

deps/LLVMExtra/lib/llvm-api.cpp

@@ -414,3 +414,7 @@ void LLVMCloneFunctionInto(LLVMValueRef NewFunc, LLVMValueRef OldFunc,
                     TypeMapper ? &TheTypeRemapper : nullptr,
                     Materializer ? &TheMaterializer : nullptr);
 }
+
+void LLVMFunctionDeleteBody(LLVMValueRef Func) {
+    unwrap<Function>(Func)->deleteBody();
+}

lib/libLLVM_extra.jl

@@ -334,3 +334,7 @@ end
 function LLVMCloneFunctionInto(NewFunc, OldFunc, ValueMap, ValueMapElements, Changes, NameSuffix, TypeMapper, TypeMapperData, Materializer, MaterializerData)
     ccall((:LLVMCloneFunctionInto, libLLVMExtra), Cvoid, (LLVMValueRef, LLVMValueRef, Ptr{LLVMValueRef}, Cuint, LLVMCloneFunctionChangeType, Cstring, Ptr{Cvoid}, Ptr{Cvoid}, Ptr{Cvoid}, Ptr{Cvoid}), NewFunc, OldFunc, ValueMap, ValueMapElements, Changes, NameSuffix, TypeMapper, TypeMapperData, Materializer, MaterializerData)
 end
+
+function LLVMFunctionDeleteBody(Func)
+    ccall((:LLVMFunctionDeleteBody, libLLVMExtra), Cvoid, (LLVMValueRef,), Func)
+end

src/core/function.jl

@@ -10,6 +10,8 @@ identify(::Type{Value}, ::Val{API.LLVMFunctionValueKind}) = Function
 Function(mod::Module, name::String, ft::FunctionType) =
     Function(API.LLVMAddFunction(mod, name, ft))
 
+Base.empty!(f::Function) = API.LLVMFunctionDeleteBody(f)
+
 unsafe_delete!(::Module, f::Function) = API.LLVMDeleteFunction(f)
 
 function personality(f::Function)

test/core.jl

@@ -1018,6 +1018,9 @@ LLVM.Module("SomeModule"; ctx) do mod
 
         @test collect(bbs) == [entrybb]
     end
+
+    empty!(fn)
+    @test isempty(blocks(fn))
 end
 end