vec fixes

JuliaSIMD · Oct 11, 2021 · 25f85c3 · 25f85c3 · chriselrod · Oct 12, 2021
1 parent fe45818
commit 25f85c3
Show file tree

Hide file tree

Showing 2 changed files with 6 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.2.4"
+version = "0.2.5"
 
 [deps]
 ArrayInterface = "4fba245c-0d91-5ea0-9b3e-6abc04ee57a9"

diff --git a/src/views.jl b/src/views.jl
@@ -112,9 +112,13 @@ end
 end
 @inline Base.SubArray(A::AbstractStrideArray, i::Tuple{Vararg{<:Union{Integer,AbstractRange,Colon},K}}) where {K} = view(A, i...)
 
+@inline Base.vec(A::PtrArray{S,D,T,1,C,0}) where {S,D,T,C} = A
 @inline function Base.vec(A::PtrArray{S,D,T,N,C,0}) where {S,D,T,N,C}
   @assert all(D) "All dimensions must be dense for a vec view. Try `vec(copy(A))` instead."
-  sp = StridedPointer(pointer(A), (static_sizeof(T),), (One(),))
+  si = StrideIndex{1,(1,),1}((static_sizeof(T),), (One(),))
+  sp = stridedpointer(pointer(A), si)
   PtrArray(sp, (static_length(A),), Val((true,)))
 end
+@inline Base.vec(A::StaticStrideArray{S,D,T,1}) where {S,D,T} = A
+@inline Base.vec(A::StaticStrideArray{S,D,T,N}) where {S,D,T,N} = StrideArray(vec(PtrArray(A)), A)