Extend vsum workaround to additional collection types

brenhinkeller · web-flow · commit e5cb5bdd47b9 · 2024-06-19T16:07:14.000-04:00
diff --git 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)

Original file line number	Diff line number	Diff line change
`@@ -24,7 +24,8 @@ julia> vsum(A, dims=2)`
`24`	`24`	`7`
`25`	`25`	```
`26`	`26`	`"""`
`27`		`-vsum(A::StridedArray; dim=:, dims=:, multithreaded=False()) = _vsum(A, dim, dims, multithreaded)`
	`27`	`+vsum(A::AbstractArray{N,T}; dim=:, dims=:, multithreaded=False()) where {N,T}= _vsum(A, dim, dims, multithreaded)`
	`28`	`+vsum(A::NTuple{N,T}; dim=:, dims=:, multithreaded=False()) where {N,T} = _vsum(A, dim, dims, multithreaded)`
`28`	`29`	`_vsum(A, ::Colon, ::Colon, multithreaded) = _vsum(A, :, multithreaded)`
`29`	`30`	`_vsum(A, ::Colon, region, multithreaded) = _vsum(A, region, multithreaded)`
`30`	`31`	`_vsum(A, region, ::Colon, multithreaded) = reducedims(_vsum(A, region, multithreaded), region)`