Cast output of cholesky to input dtype

Author: jessegrabowski

pytensor/tensor/slinalg.py

@@ -56,7 +56,7 @@ def make_node(self, x):
     def perform(self, node, inputs, outputs):
         (x,) = inputs
         (z,) = outputs
-
+        input_dtype = x.dtype
         try:
             if x.flags["C_CONTIGUOUS"] and self.overwrite_a:
                 # Inputs to the LAPACK functions need to be exactly as expected for overwrite_a to work correctly,
@@ -74,7 +74,7 @@ def perform(self, node, inputs, outputs):
                 raise
             else:
                 x = np.full_like(x, np.nan)
-        z[0] = x
+        z[0] = x.astype(input_dtype)
 
     def L_op(self, inputs, outputs, gradients):
         """