We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 30198d0 commit c1567c3Copy full SHA for c1567c3
tests/tensor/rewriting/test_linalg.py
@@ -921,9 +921,9 @@ def test_dot_kron_rewrite():
921
assert not any(isinstance(node.op, KroneckerProduct) for node in nodes)
922
923
# NUMERIC VALUE TEST
924
- a_test = np.random.rand(m, n)
925
- b_test = np.random.rand(p, q)
926
- c_test = np.random.rand(n * q, 1)
+ a_test = np.random.rand(m, n).astype(config.floatX)
+ b_test = np.random.rand(p, q).astype(config.floatX)
+ c_test = np.random.rand(n * q, 1).astype(config.floatX)
927
out_direct_val = np.kron(a_test, b_test) @ c_test
928
out_clever_val = f_direct_rewritten(a_test, b_test, c_test)
929
assert_allclose(
0 commit comments