Skip to content

Commit 22578f3

Browse files
Remove benchmark
1 parent 6becc7d commit 22578f3

File tree

1 file changed

+0
-26
lines changed

1 file changed

+0
-26
lines changed

tests/tensor/test_slinalg.py

Lines changed: 0 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -1087,29 +1087,3 @@ def test_banded_dot(A_shape, kl, ku):
10871087
out_val, out_2_val = fn(A_val, x_val)
10881088

10891089
np.testing.assert_allclose(out_val, out_2_val)
1090-
1091-
1092-
@pytest.mark.parametrize("op", ["dot", "banded_dot"], ids=str)
1093-
@pytest.mark.parametrize(
1094-
"A_shape",
1095-
[(10, 10), (100, 100), (1000, 1000), (10_000, 10_000)],
1096-
ids=["10", "100", "1000", "10_000"],
1097-
)
1098-
def test_banded_dot_perf(op, A_shape, benchmark):
1099-
rng = np.random.default_rng()
1100-
1101-
A_val = _make_banded_A(rng.normal(size=A_shape), kl=1, ku=1).astype(config.floatX)
1102-
x_val = rng.normal(size=(A_shape[-1],)).astype(config.floatX)
1103-
1104-
A = pt.tensor("A", shape=A_val.shape, dtype=A_val.dtype)
1105-
x = pt.tensor("x", shape=x_val.shape, dtype=x_val.dtype)
1106-
1107-
if op == "dot":
1108-
f = pt.dot
1109-
elif op == "banded_dot":
1110-
f = functools.partial(banded_dot, lower_diags=1, upper_diags=1)
1111-
1112-
res = f(A, x)
1113-
fn = function([A, x], res, trust_input=True)
1114-
1115-
benchmark(fn, A_val, x_val)

0 commit comments

Comments
 (0)