o i @s\ddlZddlZddlmZddlmZddlmZddlmZe d  d ddZ dS) N)cublas)device)_util)_uarraysolve_triangularFc Cst||t|jdks|jd|jdkrtdt|t|kr'td|jjdvr1|j}nt|jjd}t j ||dd d }t j ||d| d }|rp|jj dkr_t | s_td |jj dkrpt | sptd |jdkrz|jdfn|j\}} t} |dkrtj} n|d krtj} n |dkrtj} ntj} tj d|d } |rtj} ntj} |dkrtj}n|dkrtj}n|dkrtj}|rtj}ntj}| | tj| |||| | jj |j j!||j j!| |S)aSolve the equation a x = b for x, assuming a is a triangular matrix. Args: a (cupy.ndarray): The matrix with dimension ``(M, M)``. b (cupy.ndarray): The matrix with dimension ``(M,)`` or ``(M, N)``. lower (bool): Use only data contained in the lower triangle of ``a``. Default is to use upper triangle. trans (0, 1, 2, 'N', 'T' or 'C'): Type of system to solve: - *'0'* or *'N'* -- :math:`a x = b` - *'1'* or *'T'* -- :math:`a^T x = b` - *'2'* or *'C'* -- :math:`a^H x = b` unit_diagonal (bool): If ``True``, diagonal elements of ``a`` are assumed to be 1 and will not be referenced. overwrite_b (bool): Allow overwriting data in b (may enhance performance) check_finite (bool): Whether to check that the input matrices contain only finite numbers. Disabling may give a performance gain, but may result in problems (crashes, non-termination) if the inputs do contain infinities or NaNs. Returns: cupy.ndarray: The matrix with dimension ``(M,)`` or ``(M, N)``. .. seealso:: :func:`scipy.linalg.solve_triangular` rzexpected square matrixzincompatible dimensionsfdfFF)dtypeordercopyz#array must not contain infs or NaNsd)r NTC)"r_assert_cupy_arraylenshape ValueErrorr charnumpy promote_typescupyarraykindisfiniteallndimsizerget_cublas_handlerstrsmdtrsmctrsmztrsmCUBLAS_FILL_MODE_LOWERCUBLAS_FILL_MODE_UPPER CUBLAS_OP_N CUBLAS_OP_T CUBLAS_OP_CCUBLAS_DIAG_UNITCUBLAS_DIAG_NON_UNITCUBLAS_SIDE_LEFTctypesdataptr)abtranslower unit_diagonal overwrite_b check_finiter mn cublas_handletrsmoneuplodiagr?a/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/cupyx/scipy/linalg/_solve_triangular.pyr s^ !" )rFFFF) rr cupy.cudarr cupy.linalgrcupyx.scipy.linalgr implementsrr?r?r?r@s