Fix dot tests.

JuliaSIMD · Jan 25, 2021 · 5cb555e · 5cb555e · chriselrod · Jan 25, 2021
1 parent 8b2fceb
commit 5cb555e
Showing 1 changed file with 17 additions and 2 deletions.
diff --git a/test/dot.jl b/test/dot.jl
@@ -7,7 +7,7 @@ using Test
  end)
  lsdot = LoopVectorization.LoopSet(dotq);
  @test LoopVectorization.choose_order(lsdot) == (Symbol[:i], :i, Symbol("##undefined##"), :i, 4, -1)
- function mydot(a, b)
+ function mydot(a::AbstractVector, b::AbstractVector)
  s = zero(eltype(a))
  za = OffsetArray(a, OffsetArrays.Origin(0))
  zb = OffsetArray(b, OffsetArrays.Origin(0))
@@ -16,7 +16,7 @@ using Test
  end
  s
  end
- function mydotavx(a, b)
+ function mydotavx(a::AbstractVector, b::AbstractVector)
  s = zero(eltype(a))
  za = OffsetArray(a, OffsetArrays.Origin(0))
  zb = OffsetArray(b, OffsetArrays.Origin(0))
@@ -25,6 +25,21 @@ using Test
  end
  s
  end
+ @test LoopVectorization.ArrayInterface.static_step(LoopVectorization.CloseOpen(-5,10)) === LoopVectorization.One()
+ function mydot(a, b)
+ s = zero(eltype(a))
+ @inbounds @simd for i ∈ eachindex(a,b)
+ s += a[i]*b[i]
+ end
+ s
+ end
+ function mydotavx(a, b)
+ s = zero(eltype(a))
+ @avx for i ∈ eachindex(a,b)
+ s += a[i]*b[i]
+ end
+ s
+ end
 
  function mydot_avx(a, b)
  s = zero(eltype(a))