99
1010# dnrm2
1111
12- dnrm2_sizes = [100 , 1000 ]
12+ dnrm2_sizes = [100 , 200 , 400 , 600 , 800 , 1000 ]
1313
1414def 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
3838def 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
6262def 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
8888def 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
114114def 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
140140def 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
204204def run_dsyev (a , lwork , func ):
0 commit comments