@@ -6,8 +6,8 @@ using Adapt
66# # essential types
77
88export AbstractGPUArray, AbstractGPUVector, AbstractGPUMatrix, AbstractGPUVecOrMat,
9- WrappedGPUArray, AnyGPUArray, AbstractGPUArrayStyle,
10- AnyGPUArray, AnyGPUVector, AnyGPUMatrix
9+ WrappedGPUArray, AnyGPUArray, AbstractGPUArrayStyle,
10+ AnyArray, AnyGPUArray, AnyGPUVector, AnyGPUMatrix, AnyGPUVecOrMat
1111
1212"""
1313 AbstractGPUArray{T, N} <: DenseArray{T, N}
@@ -25,8 +25,10 @@ const AbstractGPUVecOrMat{T} = Union{AbstractGPUArray{T, 1}, AbstractGPUArray{T,
2525# convenience aliases for working with wrapped arrays
2626const WrappedGPUArray{T,N} = WrappedArray{T,N,AbstractGPUArray,AbstractGPUArray{T,N}}
2727const AnyGPUArray{T,N} = Union{AbstractGPUArray{T,N}, WrappedGPUArray{T,N}}
28+ const AnyArray{T,N} = Union{AbstractArray{T,N}, WrappedArray{T,N, AbstractArray,AbstractArray{T,N}}}
2829const AnyGPUVector{T} = AnyGPUArray{T, 1 }
2930const AnyGPUMatrix{T} = AnyGPUArray{T, 2 }
31+ const AnyGPUVecOrMat{T} = Union{AnyGPUArray{T, 1 }, AnyGPUArray{T, 2 }}
3032
3133# # broadcasting
3234
0 commit comments