restore the problem sizes for codspeed benchmarks

This commit is contained in:
Evgeni Burovski 2024-06-24 16:47:20 +03:00
parent 37a854718b
commit 400cf9f63d
1 changed files with 7 additions and 7 deletions

View File

@ -15,7 +15,7 @@ dtype_map = {
# dnrm2 # dnrm2
dnrm2_sizes = [100, 200, 400, 600, 800, 1000] dnrm2_sizes = [100, 1000]
def run_dnrm2(n, x, incx, func): def run_dnrm2(n, x, incx, func):
res = func(x, n, incx=incx) res = func(x, n, incx=incx)
@ -35,7 +35,7 @@ def test_nrm2(benchmark, n, variant):
# ddot # ddot
ddot_sizes = [100, 200, 400, 600, 800, 1000] ddot_sizes = [100, 1000]
def run_ddot(x, y, func): def run_ddot(x, y, func):
res = func(x, y) res = func(x, y)
@ -54,7 +54,7 @@ def test_dot(benchmark, n):
# daxpy # daxpy
daxpy_sizes = [100, 200, 400, 600, 800, 1000] daxpy_sizes = [100, 1000]
def run_daxpy(x, y, func): def run_daxpy(x, y, func):
res = func(x, y, a=2.0) res = func(x, y, a=2.0)
@ -79,7 +79,7 @@ def test_daxpy(benchmark, n, variant):
# dgemm # dgemm
gemm_sizes = [100, 200, 400, 600, 800, 1000] gemm_sizes = [100, 1000]
def run_gemm(a, b, c, func): def run_gemm(a, b, c, func):
alpha = 1.0 alpha = 1.0
@ -102,7 +102,7 @@ def test_gemm(benchmark, n, variant):
# dsyrk # dsyrk
syrk_sizes = [100, 200, 400, 600, 800, 1000] syrk_sizes = [100, 1000]
def run_syrk(a, c, func): def run_syrk(a, c, func):
@ -126,7 +126,7 @@ def test_syrk(benchmark, n, variant):
# linalg.solve # linalg.solve
gesv_sizes = [100, 200, 400, 600, 800, 1000] gesv_sizes = [100, 1000]
def run_gesv(a, b, func): def run_gesv(a, b, func):
@ -187,7 +187,7 @@ def test_gesdd(benchmark, mn, variant):
# linalg.eigh # linalg.eigh
syev_sizes = [50, 64, 128, 200] syev_sizes = [50, 200]
def run_syev(a, lwork, func): def run_syev(a, lwork, func):