Skip to content

Commit

Permalink
Fix test failure on 16 register platforms.
Browse files Browse the repository at this point in the history
  • Loading branch information
chriselrod committed Mar 24, 2020
1 parent 4d29e79 commit 9975011
Showing 1 changed file with 9 additions and 1 deletion.
10 changes: 9 additions & 1 deletion test/miscellaneous.jl
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@
s += x[m] * A[m,n] * y[n]
end);
lsdot3 = LoopVectorization.LoopSet(dot3q);
@test LoopVectorization.choose_order(lsdot3) == ([:n, :m], :m, :n, :m, Unum & -2, Tnum)
@test LoopVectorization.choose_order(lsdot3) == ([:n, :m], :m, :n, :m, Unum, Tnum)#&-2

@static if VERSION < v"1.4"
dot3(x, A, y) = dot(x, A * y)
Expand All @@ -20,6 +20,14 @@
end
s
end
function dot3avx24(x, A, y)
M, N = size(A)
s = zero(promote_type(eltype(x), eltype(A), eltype(y)))
@avx tile=(2,4) for m 1:M, n 1:N
s += x[m] * A[m,n] * y[n]
end
s
end
function dot3_avx(x, A, y)
M, N = size(A)
s = zero(promote_type(eltype(x), eltype(A), eltype(y)))
Expand Down

0 comments on commit 9975011

Please sign in to comment.