|
1 | 1 | module ArrayInterfaceGPUArrays
|
2 | 2 |
|
3 | 3 | using Adapt
|
4 |
| -using ArrayInterface |
| 4 | +using ArrayInterfaceCore |
5 | 5 | using GPUArrays
|
6 | 6 |
|
7 |
| -ArrayInterface.fast_scalar_indexing(::Type{<:GPUArrays.AbstractGPUArray}) = false |
8 |
| -@inline ArrayInterface.allowed_getindex(x::GPUArrays.AbstractGPUArray, i...) = CUDA.@allowscalar(x[i...]) |
9 |
| -@inline ArrayInterface.allowed_setindex!(x::GPUArrays.AbstractGPUArray, v, i...) = (CUDA.@allowscalar(x[i...] = v)) |
| 7 | +ArrayInterfaceCore.fast_scalar_indexing(::Type{<:GPUArrays.AbstractGPUArray}) = false |
| 8 | +@inline ArrayInterfaceCore.allowed_getindex(x::GPUArrays.AbstractGPUArray, i...) = GPUArrays.@allowscalar(x[i...]) |
| 9 | +@inline ArrayInterfaceCore.allowed_setindex!(x::GPUArrays.AbstractGPUArray, v, i...) = (GPUArrays.@allowscalar(x[i...] = v)) |
10 | 10 |
|
11 | 11 | function Base.setindex(x::GPUArrays.AbstractGPUArray, v, i::Int)
|
12 | 12 | _x = copy(x)
|
13 |
| - ArrayInterface.allowed_setindex!(_x, v, i) |
| 13 | + ArrayInterfaceCore.allowed_setindex!(_x, v, i) |
14 | 14 | return _x
|
15 | 15 | end
|
16 | 16 |
|
17 |
| -function ArrayInterface.restructure(x::GPUArrays.AbstractGPUArray, y) |
18 |
| - reshape(Adapt.adapt(ArrayInterface.parameterless_type(x), y), Base.size(x)...) |
| 17 | +function ArrayInterfaceCore.restructure(x::GPUArrays.AbstractGPUArray, y) |
| 18 | + reshape(Adapt.adapt(ArrayInterfaceCore.parameterless_type(x), y), Base.size(x)...) |
19 | 19 | end
|
20 | 20 |
|
21 |
| -function ArrayInterface.lu_instance(A::GPUArrays.AbstractGPUMatrix{T}) where {T} |
| 21 | +function ArrayInterfaceCore.lu_instance(A::GPUArrays.AbstractGPUMatrix{T}) where {T} |
22 | 22 | qr(similar(A, 1, 1))
|
23 | 23 | end
|
24 | 24 |
|
|
0 commit comments