Skip to content

Commit e7f4cf9

Browse files
committed
Fix indices dtype in numba dispatch of LU
1 parent 4e42b84 commit e7f4cf9

File tree

1 file changed

+1
-1
lines changed
  • pytensor/link/numba/dispatch/linalg/decomposition

1 file changed

+1
-1
lines changed

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ def _lu_factor_to_lu(a, dtype, overwrite_a):
3030
# Fortran is 1 indexed, so we need to subtract 1 from the IPIV array
3131
IPIV = IPIV - 1
3232
p_inv = _pivot_to_permutation(IPIV, dtype=dtype)
33-
perm = np.argsort(p_inv)
33+
perm = np.argsort(p_inv).astype("int32")
3434

3535
return perm, L, U
3636

0 commit comments

Comments
 (0)