15
15
16
16
# dnrm2
17
17
18
- dnrm2_sizes = [100 , 200 , 400 , 600 , 800 , 1000 ]
18
+ dnrm2_sizes = [100 , 1000 ]
19
19
20
20
def run_dnrm2 (n , x , incx , func ):
21
21
res = func (x , n , incx = incx )
@@ -35,7 +35,7 @@ def test_nrm2(benchmark, n, variant):
35
35
36
36
# ddot
37
37
38
- ddot_sizes = [100 , 200 , 400 , 600 , 800 , 1000 ]
38
+ ddot_sizes = [100 , 1000 ]
39
39
40
40
def run_ddot (x , y , func ):
41
41
res = func (x , y )
@@ -54,7 +54,7 @@ def test_dot(benchmark, n):
54
54
55
55
# daxpy
56
56
57
- daxpy_sizes = [100 , 200 , 400 , 600 , 800 , 1000 ]
57
+ daxpy_sizes = [100 , 1000 ]
58
58
59
59
def run_daxpy (x , y , func ):
60
60
res = func (x , y , a = 2.0 )
@@ -79,7 +79,7 @@ def test_daxpy(benchmark, n, variant):
79
79
80
80
# dgemm
81
81
82
- gemm_sizes = [100 , 200 , 400 , 600 , 800 , 1000 ]
82
+ gemm_sizes = [100 , 1000 ]
83
83
84
84
def run_gemm (a , b , c , func ):
85
85
alpha = 1.0
@@ -102,7 +102,7 @@ def test_gemm(benchmark, n, variant):
102
102
103
103
# dsyrk
104
104
105
- syrk_sizes = [100 , 200 , 400 , 600 , 800 , 1000 ]
105
+ syrk_sizes = [100 , 1000 ]
106
106
107
107
108
108
def run_syrk (a , c , func ):
@@ -126,7 +126,7 @@ def test_syrk(benchmark, n, variant):
126
126
127
127
# linalg.solve
128
128
129
- gesv_sizes = [100 , 200 , 400 , 600 , 800 , 1000 ]
129
+ gesv_sizes = [100 , 1000 ]
130
130
131
131
132
132
def run_gesv (a , b , func ):
@@ -187,7 +187,7 @@ def test_gesdd(benchmark, mn, variant):
187
187
188
188
# linalg.eigh
189
189
190
- syev_sizes = [50 , 64 , 128 , 200 ]
190
+ syev_sizes = [50 , 200 ]
191
191
192
192
193
193
def run_syev (a , lwork , func ):
0 commit comments