Skip to content

Commit 9911a94

Browse files
committed
ENH: run more problem sizes
1 parent 82cc23c commit 9911a94

1 file changed

Lines changed: 7 additions & 7 deletions

File tree

benchmarks/benchmarks.py

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

1010
# dnrm2
1111

12-
dnrm2_sizes = [100, 1000]
12+
dnrm2_sizes = [100, 200, 400, 600, 800, 1000]
1313

1414
def run_dnrm2(n, x, incx, func):
1515
res = func(x, n, incx=incx)
@@ -33,7 +33,7 @@ def time_dnrm2(self, n, variant):
3333

3434
# ddot
3535

36-
ddot_sizes = [100, 1000]
36+
ddot_sizes = [100, 200, 400, 600, 800, 1000]
3737

3838
def run_ddot(x, y, func):
3939
res = func(x, y)
@@ -57,7 +57,7 @@ def time_ddot(self, n):
5757

5858
# daxpy
5959

60-
daxpy_sizes = [100, 1000]
60+
daxpy_sizes = [100, 200, 400, 600, 800, 1000]
6161

6262
def run_daxpy(x, y, func):
6363
res = func(x, y, a=2.0)
@@ -83,7 +83,7 @@ def time_daxpy(self, n, variant):
8383

8484
# dgemm
8585

86-
gemm_sizes = [100, 1000]
86+
gemm_sizes = [100, 200, 400, 600, 800, 1000]
8787

8888
def run_dgemm(a, b, c, func):
8989
alpha = 1.0
@@ -108,7 +108,7 @@ def time_dgemm(self, n, variant):
108108

109109
# dsyrk
110110

111-
syrk_sizes = [100, 1000]
111+
syrk_sizes = [100, 200, 400, 600, 800, 1000]
112112

113113

114114
def run_dsyrk(a, c, func):
@@ -134,7 +134,7 @@ def time_dsyrk(self, n, variant):
134134

135135
# linalg.solve
136136

137-
dgesv_sizes = [100, 1000]
137+
dgesv_sizes = [100, 200, 400, 600, 800, 1000]
138138

139139

140140
def run_dgesv(a, b, func):
@@ -198,7 +198,7 @@ def time_dgesdd(self, mn, variant):
198198

199199
# linalg.eigh
200200

201-
dsyev_sizes = [50, 200]
201+
dsyev_sizes = [50, 64, 128, 200]
202202

203203

204204
def run_dsyev(a, lwork, func):

0 commit comments

Comments
 (0)