1515
1616# dnrm2
1717
18- dnrm2_sizes = [100 , 200 , 400 , 600 , 800 , 1000 ]
18+ dnrm2_sizes = [100 , 1000 ]
1919
2020def 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
4040def 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
5959def 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
8484def 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
108108def 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
132132def 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
193193def run_syev (a , lwork , func ):
0 commit comments