diff --git a/Project.toml b/Project.toml index 31f4460..ad1d3c1 100644 --- a/Project.toml +++ b/Project.toml @@ -1,7 +1,7 @@ name = "VectorizedStatistics" uuid = "3b853605-1c98-4422-8364-4bd93ee0529e" authors = ["C. Brenhin Keller", "Chris Elrod"] -version = "0.5.8" +version = "0.5.9" [deps] LoopVectorization = "bdcacae8-1622-11e9-2a5c-532679323890" diff --git a/src/vsum.jl b/src/vsum.jl index f4edc58..780485f 100644 --- a/src/vsum.jl +++ b/src/vsum.jl @@ -24,7 +24,8 @@ julia> vsum(A, dims=2) 7 ``` """ -vsum(A::StridedArray; dim=:, dims=:, multithreaded=False()) = _vsum(A, dim, dims, multithreaded) +vsum(A::AbstractArray{N,T}; dim=:, dims=:, multithreaded=False()) where {N,T}= _vsum(A, dim, dims, multithreaded) +vsum(A::NTuple{N,T}; dim=:, dims=:, multithreaded=False()) where {N,T} = _vsum(A, dim, dims, multithreaded) _vsum(A, ::Colon, ::Colon, multithreaded) = _vsum(A, :, multithreaded) _vsum(A, ::Colon, region, multithreaded) = _vsum(A, region, multithreaded) _vsum(A, region, ::Colon, multithreaded) = reducedims(_vsum(A, region, multithreaded), region)