@@ -6,8 +6,8 @@ using Adapt
6
6
# # essential types
7
7
8
8
export AbstractGPUArray, AbstractGPUVector, AbstractGPUMatrix, AbstractGPUVecOrMat,
9
- WrappedGPUArray, AnyGPUArray, AbstractGPUArrayStyle,
10
- AnyGPUArray, AnyGPUVector, AnyGPUMatrix
9
+ WrappedGPUArray, AnyGPUArray, AbstractGPUArrayStyle,
10
+ AnyArray, AnyGPUArray, AnyGPUVector, AnyGPUMatrix, AnyGPUVecOrMat
11
11
12
12
"""
13
13
AbstractGPUArray{T, N} <: DenseArray{T, N}
@@ -25,8 +25,10 @@ const AbstractGPUVecOrMat{T} = Union{AbstractGPUArray{T, 1}, AbstractGPUArray{T,
25
25
# convenience aliases for working with wrapped arrays
26
26
const WrappedGPUArray{T,N} = WrappedArray{T,N,AbstractGPUArray,AbstractGPUArray{T,N}}
27
27
const 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}}}
28
29
const AnyGPUVector{T} = AnyGPUArray{T, 1 }
29
30
const AnyGPUMatrix{T} = AnyGPUArray{T, 2 }
31
+ const AnyGPUVecOrMat{T} = Union{AnyGPUArray{T, 1 }, AnyGPUArray{T, 2 }}
30
32
31
33
# # broadcasting
32
34
0 commit comments