Quick workaround for vsum issue

brenhinkeller · brenhinkeller · commit 4ec1e74d6bde · 2023-12-05T17:54:16.000-05:00
diff --git a/src/VectorizedStatistics.jl b/src/VectorizedStatistics.jl
@@ -1,6 +1,7 @@
 module VectorizedStatistics
 
     using LoopVectorization, Static
+    import LoopVectorization.vsum # Will add more specific method for ::StridedArray
 
     const IntOrStaticInt = Union{Integer, StaticInt}
     _dim(::Type{StaticInt{N}}) where {N} = N::Int
diff --git a/src/vsum.jl b/src/vsum.jl
@@ -24,7 +24,7 @@ julia> vsum(A, dims=2)
  7
 ```
 """
-vsum(A; dim=:, dims=:, multithreaded=False()) = _vsum(A, dim, dims, multithreaded)
+vsum(A::StridedArray; dim=:, dims=:, multithreaded=False()) = _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,7 @@ julia> vsum(A, dims=2)`
`24`	`24`	`7`
`25`	`25`	```
`26`	`26`	`"""`
`27`		`-vsum(A; dim=:, dims=:, multithreaded=False()) = _vsum(A, dim, dims, multithreaded)`
	`27`	`+vsum(A::StridedArray; dim=:, dims=:, multithreaded=False()) = _vsum(A, dim, dims, multithreaded)`
`28`	`28`	`_vsum(A, ::Colon, ::Colon, multithreaded) = _vsum(A, :, multithreaded)`
`29`	`29`	`_vsum(A, ::Colon, region, multithreaded) = _vsum(A, region, multithreaded)`
`30`	`30`	`_vsum(A, region, ::Colon, multithreaded) = reducedims(_vsum(A, region, multithreaded), region)`