Revert "Update LuxDeviceUtils.jl"

This reverts commit 023fb2a.
LuxDL · Jun 26, 2023 · 983e735 · 983e735 · avik-pal · Jun 26, 2023
1 parent 023fb2a
commit 983e735
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/LuxDeviceUtils.jl b/src/LuxDeviceUtils.jl
@@ -224,12 +224,12 @@ struct LuxCUDAAdaptor <: AbstractLuxDeviceAdaptor end
 struct LuxAMDGPUAdaptor <: AbstractLuxDeviceAdaptor end
 struct LuxMetalAdaptor <: AbstractLuxDeviceAdaptor end
 
-function adapt_structure(::LuxCPUAdaptor,
+function adapt_storage(::LuxCPUAdaptor,
     x::Union{AbstractRange, SparseArrays.AbstractSparseArray})
     return x
 end
-adapt_structure(::LuxCPUAdaptor, x::AbstractArray) = adapt(Array, x)
-adapt_structure(::LuxCPUAdaptor, rng::AbstractRNG) = rng
+adapt_storage(::LuxCPUAdaptor, x::AbstractArray) = adapt(Array, x)
+adapt_storage(::LuxCPUAdaptor, rng::AbstractRNG) = rng
 
 _isbitsarray(::AbstractArray{<:Number}) = true
 _isbitsarray(::AbstractArray{T}) where {T} = isbitstype(T)