Fix support for indexing ranges and add tests.

Author: chriselrod

Manifest.toml

@@ -49,9 +49,9 @@ uuid = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
 
 [[SIMDPirates]]
 deps = ["VectorizationBase"]
-git-tree-sha1 = "a27b812034efdb062ec3d9e787bc299510057b3d"
+git-tree-sha1 = "8f1ed076d4e90047cf263b3416daa761081ca32c"
 uuid = "21efa798-c60a-11e8-04d3-e1a92915a26a"
-version = "0.3.5"
+version = "0.3.6"
 
 [[SLEEFPirates]]
 deps = ["Libdl", "SIMDPirates", "VectorizationBase"]

Project.toml

@@ -12,7 +12,7 @@ VectorizationBase = "3d5dd08c-fd9d-11e8-17fa-ed2836048c2f"
 
 [compat]
 Parameters = "0"
-SIMDPirates = "0.3.5, 0.4, 0.5"
+SIMDPirates = "0.3.6, 0.4, 0.5"
 SLEEFPirates = "0.3.2, 0.4, 0.5"
 VectorizationBase = "0.2.5, 0.3, 0.4"
 julia = "1.1"

README.md

@@ -7,7 +7,8 @@
 [![Codecov](https://codecov.io/gh/chriselrod/LoopVectorization.jl/branch/master/graph/badge.svg)](https://codecov.io/gh/chriselrod/LoopVectorization.jl)
 
 ## Installation
-```
+
+```julia
 using Pkg
 Pkg.add("LoopVectorization")
 ```

src/reconstruct_loopset.jl

@@ -104,6 +104,11 @@ function add_mref!(ls::LoopSet, ars::ArrayRefStruct, arraysymbolinds::Vector{Sym
     pushpreamble!(ls, Expr(:(=), vptr(ar), LoopValue()))
     ar
 end
+function add_mref!(ls::LoopSet, ars::ArrayRefStruct, arraysymbolinds::Vector{Symbol}, opsymbols::Vector{Symbol}, i::Int, ::Type{<:AbstractRange{T}}) where {T}
+    ar = ArrayReferenceMeta(ls, ars, arraysymbolinds, opsymbols, Symbol(""), gensym())
+    pushpreamble!(ls, Expr(:(=), vptr(ar), Expr(:macrocall, Symbol("@inbounds"), LineNumberNode(@__LINE__, @__FILE__), Expr(:ref, :vargs, i))))
+    ar
+end
 
 
 
@@ -200,6 +205,7 @@ end
 # elbytes(::VectorizationBase.AbstractPointer{T}) where {T} = sizeof(T)::Int
 typeeltype(::Type{P}) where {T,P<:VectorizationBase.AbstractPointer{T}} = T
 typeeltype(::Type{LoopValue}) = Int8
+typeeltype(::Type{<:AbstractRange{T}}) where {T} = T
 
 function add_array_symbols!(ls::LoopSet, arraysymbolinds::Vector{Symbol}, offset::Int)
     for (i,as) ∈ enumerate(arraysymbolinds)

test/runtests.jl

@@ -959,6 +959,21 @@ end
                 x[i] = a
             end
         end
+
+        function mysumavx(x)
+            s = zero(eltype(x))
+            @avx for i ∈ eachindex(x)
+                s += x[i]
+            end
+            s
+        end
+        function mysum_avx(x)
+            s = zero(eltype(x))
+            @_avx for i ∈ eachindex(x)
+                s += x[i]
+            end
+            s
+        end
 
         for T ∈ (Float32, Float64)
             @show T, @__LINE__
@@ -1101,6 +1116,19 @@ end
             @avx x .= 34.242;
             fill!(q2, 34.242)
             @test x == q2
+
+            s = sum(x)
+            @test s ≈ mysumavx(x)
+            @test s ≈ mysum_avx(x)
+            r = T == Float32 ? (Int32(-10):Int32(234)) : -10:234
+            s = sum(r)
+            @test s ≈ mysumavx(r)
+            @test s ≈ mysum_avx(r)
+            r = T(-10):T(2.3):T(1000)
+            s = T == Float32 ? sum(collect(r)) : sum(r)
+            @test s ≈ mysumavx(r)
+            @test s ≈ mysum_avx(r)
+            
         end
 end