Add KernelAbstractions.jl unsafe_free! (#1863)

Co-authored-by: Tim Besard <tim.besard@gmail.com>

Author: pxl-th

Project.toml

@@ -42,7 +42,7 @@ CUDA_Runtime_jll = "0.6"
 ExprTools = "0.1"
 GPUArrays = "8.6"
 GPUCompiler = "0.19"
-KernelAbstractions = "0.9"
+KernelAbstractions = "0.9.2"
 LLVM = "5"
 Preferences = "1"
 Random123 = "1.2"

src/CUDAKernels.jl

@@ -13,6 +13,7 @@ CUDABackend(;prefer_blocks=false, always_inline=false) = CUDABackend(prefer_bloc
 
 export CUDABackend
 
+KernelAbstractions.unsafe_free!(x::CUDA.CuArray) = CUDA.unsafe_free!(x)
 KernelAbstractions.allocate(::CUDABackend, ::Type{T}, dims::Tuple) where T = CUDA.CuArray{T}(undef, dims)
 KernelAbstractions.zeros(::CUDABackend, ::Type{T}, dims::Tuple) where T = CUDA.zeros(T, dims)
 KernelAbstractions.ones(::CUDABackend, ::Type{T}, dims::Tuple) where T = CUDA.ones(T, dims)