float32 slinalg

Author: jessegrabowski

tests/tensor/test_slinalg.py

@@ -698,7 +698,10 @@ def test_lu_solve(self, b_shape: tuple[int], trans):
         A = pt.tensor("A", shape=(5, 5))
         b = pt.tensor("b", shape=b_shape)
 
-        A_val = rng.normal(size=(5, 5)).astype(config.floatX) + np.eye(5) * 0.5
+        A_val = (
+            rng.normal(size=(5, 5)).astype(config.floatX)
+            + np.eye(5, dtype=config.floatX) * 0.5
+        )
         b_val = rng.normal(size=b_shape).astype(config.floatX)
 
         x = self.factor_and_solve(A, b, trans=trans, sum=False)