Browse Source

restore the problem sizes for codspeed benchmarks

tags/v0.3.28^2
Evgeni Burovski 1 year ago
parent
commit
400cf9f63d
1 changed files with 7 additions and 7 deletions
  1. +7
    -7
      benchmark/pybench/benchmarks/bench_blas.py

+ 7
- 7
benchmark/pybench/benchmarks/bench_blas.py View File

@@ -15,7 +15,7 @@ dtype_map = {

# dnrm2

dnrm2_sizes = [100, 200, 400, 600, 800, 1000]
dnrm2_sizes = [100, 1000]

def run_dnrm2(n, x, incx, func):
res = func(x, n, incx=incx)
@@ -35,7 +35,7 @@ def test_nrm2(benchmark, n, variant):

# ddot

ddot_sizes = [100, 200, 400, 600, 800, 1000]
ddot_sizes = [100, 1000]

def run_ddot(x, y, func):
res = func(x, y)
@@ -54,7 +54,7 @@ def test_dot(benchmark, n):

# daxpy

daxpy_sizes = [100, 200, 400, 600, 800, 1000]
daxpy_sizes = [100, 1000]

def run_daxpy(x, y, func):
res = func(x, y, a=2.0)
@@ -79,7 +79,7 @@ def test_daxpy(benchmark, n, variant):

# dgemm

gemm_sizes = [100, 200, 400, 600, 800, 1000]
gemm_sizes = [100, 1000]

def run_gemm(a, b, c, func):
alpha = 1.0
@@ -102,7 +102,7 @@ def test_gemm(benchmark, n, variant):

# dsyrk

syrk_sizes = [100, 200, 400, 600, 800, 1000]
syrk_sizes = [100, 1000]


def run_syrk(a, c, func):
@@ -126,7 +126,7 @@ def test_syrk(benchmark, n, variant):

# linalg.solve

gesv_sizes = [100, 200, 400, 600, 800, 1000]
gesv_sizes = [100, 1000]


def run_gesv(a, b, func):
@@ -187,7 +187,7 @@ def test_gesdd(benchmark, mn, variant):

# linalg.eigh

syev_sizes = [50, 64, 128, 200]
syev_sizes = [50, 200]


def run_syev(a, lwork, func):


Loading…
Cancel
Save