@@ -1087,29 +1087,3 @@ def test_banded_dot(A_shape, kl, ku):
1087
1087
out_val , out_2_val = fn (A_val , x_val )
1088
1088
1089
1089
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