o X۷iM @s>ddlmZddlZddlmZddlZddZd dd ZdS) ) annotationsN)_utilcCs@ddlm}t|t|t|t|}|||S)aCompute the inverse of a Hermitian matrix. This function computes a inverse of a real symmetric or complex hermitian positive-definite matrix using Cholesky factorization. If matrix ``a`` is not positive definite, Cholesky factorization fails and it raises an error. Args: a (cupy.ndarray): Real symmetric or complex hermitian maxtix. Returns: cupy.ndarray: The inverse of matrix ``a``. rlapack)cupyxrr_assert_cupy_array _assert_2d_assert_stacked_squarestacked_identity_likeposv)arbrI/home/ubuntu/vllm_env/lib/python3.10/site-packages/cupyx/linalg/_solve.pyinvh s     rFTcCsddlm}|\}}|r?|rt|jddnt|jdd}t|d|d|dfs4t dt|s?t d|s_|j dkrS|j j rRtj |dd d }n |j jr_tj |d d d }|j|||d S) a Solves the linear system Ax = b using the Cholesky factorization of A. Batched input arrays are also supported. Args: c (tuple of cupy.ndarray and bool): The first tuple item is the Cholesky factor of the matrix `A`, typically obtained from `cupy.linalg.cholesky`. The second item is a bool that indicates whether the Cholesky factor is stored in the lower-triangular part (when True) or in the upper-triangular part (when False). b (cupy.ndarray): Right-hand side array. overwrite_b (bool, optional): If True, the contents of `b` may be overwritten for efficiency. Default is False. check_finite (bool, optional): If True, checks whether the input arrays contain only finite numbers. Disabling this may improve performance but can lead to crashes or non-termination if NaNs or infs are present. Default is True. Returns: cupy.ndarray: Solution to the linear system Ax = b. See Also: cupy.linalg.cho_factor: Computes the Cholesky factorization. Examples: >>> import cupy as cp >>> from cupy.linalg import cholesky, cho_solve >>> A = cp.array([[9, 3, 1, 5], ... [3, 7, 5, 1], ... [1, 5, 9, 2], ... [5, 1, 2, 6]]) >>> c= cholesky(A) >>> x = cho_solve((c, True), cp.ones(4)) >>> cp.allclose(A @ x, cp.ones(4)) True rr.z%Input array contains NaN or infinity.FT)ordercopyC)lower)rrnumpy tril_indicesshape triu_indicescupyisfiniteall ValueErrorndimflags f_contiguousasarray c_contiguouspotrs) c_and_lowerr overwrite_b check_finitercrindexesrrr cho_solve"s$ %  r,)FT) __future__rr cupy.linalgrrrr,rrrrs