Skip to content

Commit 03fe360

Browse files
Adding AnyGPUVecOrMat and AnyArray to core
1 parent 36661e3 commit 03fe360

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

lib/GPUArraysCore/src/GPUArraysCore.jl

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,8 +6,8 @@ using Adapt
66
## essential types
77

88
export 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
2626
const WrappedGPUArray{T,N} = WrappedArray{T,N,AbstractGPUArray,AbstractGPUArray{T,N}}
2727
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}}}
2829
const AnyGPUVector{T} = AnyGPUArray{T, 1}
2930
const AnyGPUMatrix{T} = AnyGPUArray{T, 2}
31+
const AnyGPUVecOrMat{T} = Union{AnyGPUArray{T, 1}, AnyGPUArray{T, 2}}
3032

3133
## broadcasting
3234

0 commit comments

Comments
 (0)