Merge pull request #139 from JuliaArrays/staticarrayviewstrides

Fix strides for view of a StaticArray.

Author: chriselrod

Project.toml

@@ -1,6 +1,6 @@
 name = "ArrayInterface"
 uuid = "4fba245c-0d91-5ea0-9b3e-6abc04ee57a9"
-version = "3.1.7"
+version = "3.1.8"
 
 [deps]
 IfElse = "615f187c-cbe4-4ef1-ba3b-2fcf58d6d173"

src/ArrayInterface.jl

@@ -926,6 +926,7 @@ function __init__()
             end
             return t
         end
+        @inline strides(B::StaticArrays.SizedArray{S,T,M,N,A}) where {S,T,M,N,A<:SubArray} = strides(B.data)
         if StaticArrays.SizedArray{Tuple{8,8},Float64,2,2} isa UnionAll
             parent_type(::Type{<:StaticArrays.SizedArray{S,T,M,N,A}}) where {S,T,M,N,A} = A
         else

test/runtests.jl

@@ -5,8 +5,10 @@ using ArrayInterface: StaticInt, True, False
 import ArrayInterface: has_sparsestruct, findstructralnz, fast_scalar_indexing, lu_instance, device, contiguous_axis, contiguous_batch_size, stride_rank, dense_dims, static
 @test ArrayInterface.ismutable(rand(3))
 
-using Aqua
-Aqua.test_all(ArrayInterface)
+if VERSION ≥ v"1.6"
+    using Aqua
+    Aqua.test_all(ArrayInterface)
+end
 
 using StaticArrays
 x = @SVector [1,2,3]
@@ -460,6 +462,10 @@ using OffsetArrays
 
     @test @inferred(ArrayInterface.indices(OffsetArray(view(PermutedDimsArray(A, (3,1,2)), 1, :, 2:4)', 3, -173),1)) === Base.Slice(ArrayInterface.OptionallyStaticUnitRange(4,6))
     @test @inferred(ArrayInterface.indices(OffsetArray(view(PermutedDimsArray(A, (3,1,2)), 1, :, 2:4)', 3, -173),2)) === Base.Slice(ArrayInterface.OptionallyStaticUnitRange(-172,-170))
+
+    Am = @MMatrix rand(2,10);
+    @test @inferred(ArrayInterface.strides(view(Am,1,:))) === (StaticInt(2),)
+
 end
 
 @testset "Static-Dynamic Size, Strides, and Offsets" begin