Not dense, CPUPointer device for SubArrays of

JuliaSIMD · Jul 11, 2023 · 7131c89 · 7131c89 · chriselrod · Jul 11, 2023
1 parent 441fa05
commit 7131c89
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "StrideArraysCore"
 uuid = "7792a7ef-975c-4747-a70f-980b88e8d1da"
 authors = ["Chris Elrod <[email protected]> and contributors"]
-version = "0.4.16"
+version = "0.4.17"
 
 [deps]
 ArrayInterface = "4fba245c-0d91-5ea0-9b3e-6abc04ee57a9"

diff --git a/src/ptr_array.jl b/src/ptr_array.jl
@@ -189,7 +189,7 @@ abstract type AbstractStrideArray{
   S<:Tuple{Vararg{Integer,N}},
   X<:Tuple{Vararg{Union{Integer,Nothing,StrideReset},N}},
   O<:Tuple{Vararg{Integer,N}}
-} <: DenseArray{T,N} end
+} <: AbstractArray{T,N} end
 abstract type AbstractPtrStrideArray{T,N,R,S,X,O} <:
               AbstractStrideArray{T,N,R,S,X,O} end
 const AbstractStrideVector{T,R,S,X,O} = AbstractStrideArray{T,1,R,S,X,O}
@@ -495,6 +495,9 @@ end
   _strides_entry(static_size(A), getfield(A, :strides), Val{R}(), valisbit(A))
 end
 ArrayInterface.device(::AbstractStrideArray) = ArrayInterface.CPUPointer()
+ArrayInterface.device(
+  ::SubArray{T,NS,AbstractPtrArray{T,N,R,S,X,O,P}}
+) where {T,NS,N,R,S,X,O,P} = ArrayInterface.CPUPointer()
 
 @generated function ArrayInterface.contiguous_axis(
   ::Type{<:AbstractStrideArray{<:Any,<:Any,R,<:Any,X}}