@@ -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