Skip to content

Commit 5cb555e

Browse files
committed
Fix dot tests.
1 parent 8b2fceb commit 5cb555e

File tree

1 file changed

+17
-2
lines changed

1 file changed

+17
-2
lines changed

test/dot.jl

Lines changed: 17 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ using Test
77
end)
88
lsdot = LoopVectorization.LoopSet(dotq);
99
@test LoopVectorization.choose_order(lsdot) == (Symbol[:i], :i, Symbol("##undefined##"), :i, 4, -1)
10-
function mydot(a, b)
10+
function mydot(a::AbstractVector, b::AbstractVector)
1111
s = zero(eltype(a))
1212
za = OffsetArray(a, OffsetArrays.Origin(0))
1313
zb = OffsetArray(b, OffsetArrays.Origin(0))
@@ -16,7 +16,7 @@ using Test
1616
end
1717
s
1818
end
19-
function mydotavx(a, b)
19+
function mydotavx(a::AbstractVector, b::AbstractVector)
2020
s = zero(eltype(a))
2121
za = OffsetArray(a, OffsetArrays.Origin(0))
2222
zb = OffsetArray(b, OffsetArrays.Origin(0))
@@ -25,6 +25,21 @@ using Test
2525
end
2626
s
2727
end
28+
@test LoopVectorization.ArrayInterface.static_step(LoopVectorization.CloseOpen(-5,10)) === LoopVectorization.One()
29+
function mydot(a, b)
30+
s = zero(eltype(a))
31+
@inbounds @simd for i eachindex(a,b)
32+
s += a[i]*b[i]
33+
end
34+
s
35+
end
36+
function mydotavx(a, b)
37+
s = zero(eltype(a))
38+
@avx for i eachindex(a,b)
39+
s += a[i]*b[i]
40+
end
41+
s
42+
end
2843

2944
function mydot_avx(a, b)
3045
s = zero(eltype(a))

0 commit comments

Comments
 (0)