提交 fd704953 authored 作者: Ricardo Vieira's avatar Ricardo Vieira 提交者: Jesse Grabowski

Fix indices dtype in numba dispatch of LU

上级 9238cc23
......@@ -30,7 +30,7 @@ def _lu_factor_to_lu(a, dtype, overwrite_a):
# Fortran is 1 indexed, so we need to subtract 1 from the IPIV array
IPIV = IPIV - 1
p_inv = _pivot_to_permutation(IPIV, dtype=dtype)
perm = np.argsort(p_inv)
perm = np.argsort(p_inv).astype("int32")
return perm, L, U
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论