Skip to content

Commit d8eecae

Browse files
committed
Define tests for the gemm/gemv benchmarks.
1 parent 0bde1c3 commit d8eecae

File tree

1 file changed

+12
-6
lines changed

1 file changed

+12
-6
lines changed

benchmark/driver.jl

Lines changed: 12 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -26,34 +26,38 @@ blastests() = [
2626
"OpenBLAS", "MKL"
2727
]
2828
function benchmark_AmulB(sizes)
29+
tests = blastests()
2930
start_workers()
3031
sm = SharedMatrix(Matrix{Float64}(undef, length(tests), length(sizes)))
3132
pmap(is -> A_mul_B_bench!(sm, is[2], is[1]), enumerate(sizes))
32-
br = BenchmarkResult(Matrix(sm), blastests(), sizes)
33+
br = BenchmarkResult(Matrix(sm), tests, sizes)
3334
stop_workers()
3435
br
3536
end
3637
function benchmark_AmulBt(sizes)
38+
tests = blastests()
3739
start_workers()
3840
sm = SharedMatrix(Matrix{Float64}(undef, length(tests), length(sizes)))
3941
pmap(is -> A_mul_Bt_bench!(sm, is[2], is[1]), enumerate(sizes))
40-
br = BenchmarkResult(Matrix(sm), blastests(), sizes)
42+
br = BenchmarkResult(Matrix(sm), tests, sizes)
4143
stop_workers()
4244
br
4345
end
4446
function benchmark_AtmulB(sizes)
47+
tests = blastests()
4548
start_workers()
4649
sm = SharedMatrix(Matrix{Float64}(undef, length(tests), length(sizes)))
4750
pmap(is -> At_mul_B_bench!(sm, is[2], is[1]), enumerate(sizes))
48-
br = BenchmarkResult(Matrix(sm), blastests(), sizes)
51+
br = BenchmarkResult(Matrix(sm), tests, sizes)
4952
stop_workers()
5053
br
5154
end
5255
function benchmark_AtmulBt(sizes)
56+
tests = blastests()
5357
start_workers()
5458
sm = SharedMatrix(Matrix{Float64}(undef, length(tests), length(sizes)))
5559
pmap(is -> At_mul_Bt_bench!(sm, is[2], is[1]), enumerate(sizes))
56-
br = BenchmarkResult(Matrix(sm), blastests(), sizes)
60+
br = BenchmarkResult(Matrix(sm), tests, sizes)
5761
stop_workers()
5862
br
5963
end
@@ -76,18 +80,20 @@ function benchmark_selfdot(sizes)
7680
br
7781
end
7882
function benchmark_Amulvb(sizes)
83+
tests = blastests()
7984
start_workers()
8085
sm = SharedMatrix(Matrix{Float64}(undef, length(tests), length(sizes)))
8186
pmap(is -> A_mul_vb_bench!(sm, is[2], is[1]), enumerate(sizes))
82-
br = BenchmarkResult(Matrix(sm), blastests(), sizes)
87+
br = BenchmarkResult(Matrix(sm), tests, sizes)
8388
stop_workers()
8489
br
8590
end
8691
function benchmark_Atmulvb(sizes)
92+
tests = blastests()
8793
start_workers()
8894
sm = SharedMatrix(Matrix{Float64}(undef, length(tests), length(sizes)))
8995
pmap(is -> At_mul_vb_bench!(sm, is[2], is[1]), enumerate(sizes))
90-
br = BenchmarkResult(Matrix(sm), blastests(), sizes)
96+
br = BenchmarkResult(Matrix(sm), tests, sizes)
9197
stop_workers()
9298
br
9399
end

0 commit comments

Comments
 (0)