Adapt to JuliaGPU/GPUArrays.jl#567. (#2537)

Author: maleadt

Project.toml

@@ -59,7 +59,7 @@ Crayons = "4"
 DataFrames = "1"
 EnzymeCore = "0.8.2"
 ExprTools = "0.1"
-GPUArrays = "11"
+GPUArrays = "11.1"
 GPUCompiler = "0.24, 0.25, 0.26, 0.27, 1"
 KernelAbstractions = "0.9.2"
 LLVM = "9.1"

src/array.jl

@@ -84,14 +84,7 @@ mutable struct CuArray{T,N,M} <: AbstractGPUArray{T,N}
   end
 end
 
-"""
-    CUDA.unsafe_free!(a::CuArray)
-
-Release the memory of an array for reuse by future allocations. This operation is
-performed automatically by the GC when an array goes out of scope, but can be called
-earlier to reduce pressure on the memory allocator.
-"""
-unsafe_free!(xs::CuArray) = GPUArrays.unsafe_free!(xs.data)
+GPUArrays.storage(a::CuArray) = a.data
 
 
 ## alias detection