fix strides of zero-dimensional arrays

Jutho · Mar 1, 2019 · da6574d · da6574d
1 parent c9d9e41
commit da6574d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/src/stridedview.jl b/src/stridedview.jl
@@ -35,6 +35,7 @@ StridedView(a::Base.ReshapedArray) = sreshape(StridedView(a.parent), a.dims)
 # Methods for StridedView
 Base.size(a::StridedView) = a.size
 Base.strides(a::StridedView) = a.strides
+Base.stride(a::StridedView{<:Any, 0}, n::Int) = 1
 Base.stride(a::StridedView{<:Any, N}, n::Int) where N =
     (n <= N) ? a.strides[n] : a.strides[N]*a.size[N]
 offset(a::StridedView) = a.offset

diff --git a/src/unsafestridedview.jl b/src/unsafestridedview.jl
@@ -28,6 +28,7 @@ UnsafeStridedView(a::Base.ReshapedArray) = sreshape(UnsafeStridedView(a.parent),
 # Methods for UnsafeStridedView
 Base.size(a::UnsafeStridedView) = a.size
 Base.strides(a::UnsafeStridedView) = a.strides
+Base.stride(a::UnsafeStridedView{<:Any, 0}, n::Int) = 1
 Base.stride(a::UnsafeStridedView{<:Any, N}, n::Int) where {N} =
     (n <= N) ? a.strides[n] : a.strides[N]*a.size[N]
 offset(a::UnsafeStridedView) = a.offset