o 0 i@s@ddlZddlZddlmZddlmZddlmZddZdS)N)device)_util)sparsec Csddlm}t|st|}t|t||jd}|j dks*t ||kr.t d|j dks8|j dkr<|j }nt |j d}t}|j}d}d}tj||d} t dt j} |dkrd|j} n|j} | ||||jj|jjj|jjj|jjj|jj||| jj| jj | t j} | S) aESolves linear system with cholesky decomposition. Find the solution to a large, sparse, linear system of equations. The function solves ``Ax = b``. Given two-dimensional matrix ``A`` is decomposed into ``L * L^*``. Args: A (cupy.ndarray or cupyx.scipy.sparse.csr_matrix): The input matrix with dimension ``(N, N)``. Must be positive-definite input matrix. Only symmetric real matrix is supported currently. b (cupy.ndarray): Right-hand side vector. Returns: ret (cupy.ndarray): The solution vector ``x``. r)cusolverz+b must be 1-d array whose size is same as Afdg?)dtype) Zcupy_backends.cuda.libsrrZisspmatrix_csrZ csr_matrixrZ_assert_stacked_squareZ_assert_cupy_arrayshapendimlen ValueErrorr numpyZ promote_typesrZget_cusolver_sp_handlennzcupyemptyZint32Z scsrlsvcholZ dcsrlsvcholZ_descrZ descriptordataZptrZindptrindicesctypesZastypeZfloat64) Abrmr handlerZtolZreorderx singularityZ csrlsvcholrf/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/cupyx/linalg/sparse/_solve.pylschol s6        r) rrZ cupy.cudarZ cupy.linalgrZ cupyx.scipyrrrrrrs