`i ~ddlZddlZddlmZddlmZddlmZddlmZej d d dZ dS) N)cublas)device)_util)_uarraysolve_triangularFchtj||t|jdks|jd|jdkrt dt|t|krt d|jjdvr|j}ntj|jjd}tj ||dd }tj ||d| }|r|jj dkr5tj | st d |jj dkr5tj | st d |jdkr |jdfn|j\}} t!j} |dkr t$j} n2|d kr t$j} n|dkr t$j} n t$j} tj d| } |r t$j} n t$j} |dkr t$j}n%|dkr t$j}n|dkr t$j}|r t$j}n t$j}| | t$j| |||| | 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)rNTC)"r_assert_cupy_arraylenshape ValueErrorrcharnumpy 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_finitermn cublas_handletrsmoneuplodiags x/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/cupyx/scipy/linalg/_solve_triangular.pyrr sB Q""" 17||qAGAJ!'!*441222 1vvQ2333 w|t#AGL#66 1E5999A 1E OEEEA7 7<3  t}Q'7'7';';'='= 577 7 7<3  t}Q'7'7';';'='= 577 7&A++AFA;;17DAq,..M ||| #| #|| +au % % %C -,, ||" #" #"+&*Dv. t 1cjoqvz1afj!=== H)rFFFF) rr cupy.cudarr cupy.linalgrcupyx.scipy.linalgr implementsrrBrArHs &&&&&&&''?D5:Z Z Z ('Z Z Z rB