Skip to content

Commit 912a88d

Browse files
Update tests/tensor/test_slinalg.py
Co-authored-by: Ricardo Vieira <28983449+ricardoV94@users.noreply.github.com>
1 parent ae596b6 commit 912a88d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/tensor/test_slinalg.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1101,8 +1101,8 @@ def test_banded_dot_perf(A_shape, kl, ku, benchmark):
11011101
A_val = _make_banded_A(rng.normal(size=A_shape), kl=kl, ku=ku)
11021102
b_val = rng.normal(size=(A_shape[-1],))
11031103

1104-
A = pt.tensor("A", shape=A_val.shape)
1105-
b = pt.tensor("b", shape=b_val.shape)
1104+
A = pt.tensor("A", shape=A_val.shape, dtype=A_val.dtype)
1105+
b = pt.tensor("b", shape=b_val.shape, dtype=b_val.dtype)
11061106

11071107
res = banded_dot(A, b, kl, ku)
11081108
fn = function([A, b], res, trust_input=True)

0 commit comments

Comments
 (0)