Merge pull request #33 from JuliaSIMD/vsum

brenhinkeller · web-flow · commit 6626b0ee7518 · 2024-06-13T03:52:25.000-04:00
Quick workaround for `vsum` issue
diff --git 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.7"
+version = "0.5.8"
 
 [deps]
 LoopVectorization = "bdcacae8-1622-11e9-2a5c-532679323890"
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)`