|
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