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

JuliaGPU · Oct 30, 2024 · 7d23bbf · 7d23bbf
1 parent c2f0535
commit 7d23bbf
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 9 deletions.
diff --git a/Project.toml b/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"

diff --git a/src/array.jl b/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