Fix numba symmetrical solve reciprocal of condition number

Author: ricardoV94

pytensor/link/numba/dispatch/slinalg.py

@@ -654,7 +654,7 @@ def impl(
 
 def _sysv(
  A: np.ndarray, B: np.ndarray, lower: bool, overwrite_a: bool, overwrite_b: bool
-) -> tuple[np.ndarray, np.ndarray, int]:
+) -> tuple[np.ndarray, np.ndarray, np.ndarray, int]:
  """
  Placeholder for solving a linear system with a symmetric matrix; used by linalg.solve.
  """
@@ -665,7 +665,8 @@ def _sysv(
 def sysv_impl(
  A: np.ndarray, B: np.ndarray, lower: bool, overwrite_a: bool, overwrite_b: bool
 ) -> Callable[
- [np.ndarray, np.ndarray, bool, bool, bool], tuple[np.ndarray, np.ndarray, int]
+ [np.ndarray, np.ndarray, bool, bool, bool],
+ tuple[np.ndarray, np.ndarray, np.ndarray, int],
 ]:
  ensure_lapack()
  _check_scipy_linalg_matrix(A, "sysv")
@@ -741,8 +742,8 @@ def impl(
  )
 
  if B_is_1d:
- return B_copy[..., 0], IPIV, int_ptr_to_val(INFO)
- return B_copy, IPIV, int_ptr_to_val(INFO)
+ B_copy = B_copy[..., 0]
+ return A_copy, B_copy, IPIV, int_ptr_to_val(INFO)
 
  return impl
 
@@ -771,7 +772,7 @@ def impl(A: np.ndarray, ipiv: np.ndarray, anorm: float) -> tuple[np.ndarray, int
 
  N = val_to_int_ptr(_N)
  LDA = val_to_int_ptr(_N)
- UPLO = val_to_int_ptr(ord("L"))
+ UPLO = val_to_int_ptr(ord("U"))
  ANORM = np.array(anorm, dtype=dtype)
  RCOND = np.empty(1, dtype=dtype)
  WORK = np.empty(2 * _N, dtype=dtype)
@@ -844,10 +845,10 @@ def impl(
  ) -> np.ndarray:
  _solve_check_input_shapes(A, B)
 
- x, ipiv, info = _sysv(A, B, lower, overwrite_a, overwrite_b)
+ lu, x, ipiv, info = _sysv(A, B, lower, overwrite_a, overwrite_b)
  _solve_check(A.shape[-1], info)
 
- rcond, info = _sycon(A, ipiv, _xlange(A, order="I"))
+ rcond, info = _sycon(lu, ipiv, _xlange(A, order="I"))
  _solve_check(A.shape[-1], info, True, rcond)
 
  return x