Skip to content

Commit 5cedee2

Browse files
committed
Update tile sizes in a few tests for 16 register platforms.
1 parent 4d77be0 commit 5cedee2

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

test/gemm.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -294,7 +294,7 @@
294294
if LoopVectorization.VectorizationBase.REGISTER_COUNT == 32
295295
@test LoopVectorization.choose_order(lsr2amb) == ([:n, :m, :k], :k, :n, :m, 3, 6)
296296
else
297-
@test LoopVectorization.choose_order(lsr2amb) == ([:n, :m, :k], :k, :n, :m, 2, 4)
297+
@test LoopVectorization.choose_order(lsr2amb) == ([:n, :m, :k], :k, :n, :m, 1, 6)
298298
end
299299
function rank2AmulBavx!(C, Aₘ, Aₖ, B)
300300
@avx for m 1:size(C,1), n 1:size(C,2)

test/gemv.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@ using LoopVectorization
22
using Test
33

44
@testset "GEMV" begin
5-
Unum, Tnum = LoopVectorization.VectorizationBase.REGISTER_COUNT == 16 ? (3, 4) : (4, 6)
5+
Unum, Tnum = LoopVectorization.VectorizationBase.REGISTER_COUNT == 16 ? (2, 6) : (4, 6)
66
gemvq = :(for i eachindex(y)
77
yᵢ = 0.0
88
for j eachindex(x)

test/miscellaneous.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11

22
@testset "Miscellaneous" begin
3-
Unum, Tnum = LoopVectorization.VectorizationBase.REGISTER_COUNT == 16 ? (3, 4) : (4, 6)
3+
Unum, Tnum = LoopVectorization.VectorizationBase.REGISTER_COUNT == 16 ? (2, 6) : (4, 6)
44
dot3q = :(for m 1:M, n 1:N
55
s += x[m] * A[m,n] * y[n]
66
end);

0 commit comments

Comments
 (0)