.Numba linalg: handle empty inputs and dtypes carefully

Author: ricardoV94

pytensor/link/numba/dispatch/linalg/decomposition/cholesky.py

@@ -5,7 +5,6 @@
 
 from pytensor.link.numba.dispatch.linalg._LAPACK import (
     _LAPACK,
-    _get_underlying_float,
     int_ptr_to_val,
     val_to_int_ptr,
 )
@@ -26,7 +25,7 @@ def cholesky_impl(A, lower=0, overwrite_a=False, check_finite=True):
     ensure_lapack()
     _check_scipy_linalg_matrix(A, "cholesky")
     dtype = A.dtype
-    w_type = _get_underlying_float(dtype)
+
     numba_potrf = _LAPACK().numba_xpotrf(dtype)
 
     def impl(A, lower=False, overwrite_a=False, check_finite=True):
@@ -53,7 +52,7 @@ def impl(A, lower=False, overwrite_a=False, check_finite=True):
         numba_potrf(
             UPLO,
             N,
-            A_copy.view(w_type).ctypes,
+            A_copy.ctypes,
             LDA,
             INFO,
         )

pytensor/link/numba/dispatch/linalg/decomposition/lu_factor.py

@@ -8,7 +8,6 @@
 
 from pytensor.link.numba.dispatch.linalg._LAPACK import (
     _LAPACK,
-    _get_underlying_float,
     int_ptr_to_val,
     val_to_int_ptr,
 )
@@ -40,7 +39,6 @@ def getrf_impl(
     ensure_lapack()
     _check_scipy_linalg_matrix(A, "getrf")
     dtype = A.dtype
-    w_type = _get_underlying_float(dtype)
     numba_getrf = _LAPACK().numba_xgetrf(dtype)
 
     def impl(
@@ -59,7 +57,7 @@ def impl(
         IPIV = np.empty(_N, dtype=np.int32)  # type: ignore
         INFO = val_to_int_ptr(0)
 
-        numba_getrf(M, N, A_copy.view(w_type).ctypes, LDA, IPIV.ctypes, INFO)
+        numba_getrf(M, N, A_copy.ctypes, LDA, IPIV.ctypes, INFO)
 
         return A_copy, IPIV, int_ptr_to_val(INFO)