Skip to content

Commit 400cf9f

Browse files
committed
restore the problem sizes for codspeed benchmarks
1 parent 37a8547 commit 400cf9f

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

benchmark/pybench/benchmarks/bench_blas.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515

1616
# dnrm2
1717

18-
dnrm2_sizes = [100, 200, 400, 600, 800, 1000]
18+
dnrm2_sizes = [100, 1000]
1919

2020
def run_dnrm2(n, x, incx, func):
2121
res = func(x, n, incx=incx)
@@ -35,7 +35,7 @@ def test_nrm2(benchmark, n, variant):
3535

3636
# ddot
3737

38-
ddot_sizes = [100, 200, 400, 600, 800, 1000]
38+
ddot_sizes = [100, 1000]
3939

4040
def run_ddot(x, y, func):
4141
res = func(x, y)
@@ -54,7 +54,7 @@ def test_dot(benchmark, n):
5454

5555
# daxpy
5656

57-
daxpy_sizes = [100, 200, 400, 600, 800, 1000]
57+
daxpy_sizes = [100, 1000]
5858

5959
def run_daxpy(x, y, func):
6060
res = func(x, y, a=2.0)
@@ -79,7 +79,7 @@ def test_daxpy(benchmark, n, variant):
7979

8080
# dgemm
8181

82-
gemm_sizes = [100, 200, 400, 600, 800, 1000]
82+
gemm_sizes = [100, 1000]
8383

8484
def run_gemm(a, b, c, func):
8585
alpha = 1.0
@@ -102,7 +102,7 @@ def test_gemm(benchmark, n, variant):
102102

103103
# dsyrk
104104

105-
syrk_sizes = [100, 200, 400, 600, 800, 1000]
105+
syrk_sizes = [100, 1000]
106106

107107

108108
def run_syrk(a, c, func):
@@ -126,7 +126,7 @@ def test_syrk(benchmark, n, variant):
126126

127127
# linalg.solve
128128

129-
gesv_sizes = [100, 200, 400, 600, 800, 1000]
129+
gesv_sizes = [100, 1000]
130130

131131

132132
def run_gesv(a, b, func):
@@ -187,7 +187,7 @@ def test_gesdd(benchmark, mn, variant):
187187

188188
# linalg.eigh
189189

190-
syev_sizes = [50, 64, 128, 200]
190+
syev_sizes = [50, 200]
191191

192192

193193
def run_syev(a, lwork, func):

0 commit comments

Comments
 (0)