Merge pull request #320 from JuliaArrays/gpuarrayscore

Change ArrayInterfaceGPUArrays to use GPUArraysCore

Author: ChrisRackauckas

lib/ArrayInterfaceGPUArrays/Project.toml

@@ -1,16 +1,16 @@
 name = "ArrayInterfaceGPUArrays"
 uuid = "6ba088a2-8465-4c0a-af30-387133b534db"
-version = "0.1.0"
+version = "0.2.0"
 
 [deps]
 Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"
 ArrayInterfaceCore = "30b0a656-2188-435a-8636-2ec0e6a096e2"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
-GPUArrays = "0c68f7d7-f131-5f86-a1c3-88cf8149b2d7"
+GPUArraysCore = "46192b85-c4d5-4398-a991-12ede77f4527"
 
 [compat]
 Adapt = "3"
 ArrayInterfaceCore = "0.1.1"
-GPUArrays = "8"
+GPUArraysCore = "0.1"
 julia = "1.6"
 

lib/ArrayInterfaceGPUArrays/src/ArrayInterfaceGPUArrays.jl

@@ -2,27 +2,27 @@ module ArrayInterfaceGPUArrays
 
 using Adapt
 using ArrayInterfaceCore
-using GPUArrays
+using GPUArraysCore
 
-ArrayInterfaceCore.fast_scalar_indexing(::Type{<:GPUArrays.AbstractGPUArray}) = false
-@inline ArrayInterfaceCore.allowed_getindex(x::GPUArrays.AbstractGPUArray, i...) = GPUArrays.@allowscalar(x[i...])
-@inline ArrayInterfaceCore.allowed_setindex!(x::GPUArrays.AbstractGPUArray, v, i...) = (GPUArrays.@allowscalar(x[i...] = v))
+ArrayInterfaceCore.fast_scalar_indexing(::Type{<:GPUArraysCore.AbstractGPUArray}) = false
+@inline ArrayInterfaceCore.allowed_getindex(x::GPUArraysCore.AbstractGPUArray, i...) = GPUArraysCore.@allowscalar(x[i...])
+@inline ArrayInterfaceCore.allowed_setindex!(x::GPUArraysCore.AbstractGPUArray, v, i...) = (GPUArraysCore.@allowscalar(x[i...] = v))
 
-function Base.setindex(x::GPUArrays.AbstractGPUArray, v, i::Int)
+function Base.setindex(x::GPUArraysCore.AbstractGPUArray, v, i::Int)
     _x = copy(x)
     ArrayInterfaceCore.allowed_setindex!(_x, v, i)
     return _x
 end
 
-function ArrayInterfaceCore.restructure(x::GPUArrays.AbstractGPUArray, y)
+function ArrayInterfaceCore.restructure(x::GPUArraysCore.AbstractGPUArray, y)
     reshape(Adapt.adapt(ArrayInterfaceCore.parameterless_type(x), y), Base.size(x)...)
 end
 
-function ArrayInterfaceCore.lu_instance(A::GPUArrays.AbstractGPUMatrix{T}) where {T}
+function ArrayInterfaceCore.lu_instance(A::GPUArraysCore.AbstractGPUMatrix{T}) where {T}
     qr(similar(A, 1, 1))
 end
 
 # Doesn't do much, but makes a gigantic change to the dependency chain.
-# ArrayInterface.device(::Type{<:GPUArrays.AbstractGPUArray}) = ArrayInterface.GPU()
+# ArrayInterface.device(::Type{<:GPUArraysCore.AbstractGPUArray}) = ArrayInterface.GPU()
 
 end