o Õ0 iŸć@sbddlZddlZddlmZddlmZddlmZddlmZddd„Z dd d „Z dd d „Z dS)éN)Śdevice)Śruntime)Ś_util)Ś_dtypeFcCsvddlm}ddlm}|dvrtdƒ‚tj|dd\}}|j ”}|j ”} |j|d| d } |j \} } t   | |”} t   d t j ”}t ”j}|rM|j}n|j}|d krX|j}n|j}tjsĢ|jd vrgtd ƒ‚t |”}t |”}| ”}zF| ||||| || jj| || jj|” \}}t   |d”}t   |d”}| ||||| || jj| || jj||jj||jj||jj”W|  |”n|  |”wt j!j "|j|”na|dkr×|j#}|j$}n%|dkrā|j%}|j&}n|dkrķ|j'}|j(}n|dkrų|j)}|j*}ntd ƒ‚||||| | jj| | jjƒ}t   ||”}||||| | jj| | jj|jj||jjƒ t j!j "||”| j| dd| j|ddfS)Nr)Ścublas)Ścusolver)ŚLŚUz UPLO argument must be 'L' or 'U'F)Zreject_float16ŚF)ŚorderŚcopy©rZfdFDz>Only float32, float64, complex64, and complex128 are supportedŚbŚfŚdŚD)r )+Zcupy_backends.cuda.libsrrŚ ValueErrorrŚlinalg_common_typeŚcharŚlowerZastypeŚshapeŚcupyŚemptyŚnumpyZint32rZDeviceZcusolver_handleZCUSOLVER_EIG_MODE_VECTORZCUSOLVER_EIG_MODE_NOVECTORZCUBLAS_FILL_MODE_LOWERZCUBLAS_FILL_MODE_UPPERrŚis_hipŚ RuntimeErrorrZ to_cuda_dtypeZ createParamsZxsyevd_bufferSizeŚdataZptrZxsyevdŚctypesZ destroyParamsZlinalgZ3_check_cusolver_dev_info_if_synchronization_allowedZssyevd_bufferSizeZssyevdZdsyevd_bufferSizeZdsyevdZcheevd_bufferSizeZcheevdZzheevd_bufferSizeZzheevd)ŚaŚUPLOZwith_eigen_vectorZ overwrite_arrZdtypeŚv_dtypeZ real_dtypeŚw_dtypeŚvŚmZldaŚwZdev_infoŚhandleZjobzZuploZtype_vZtype_wŚparamsZwork_device_sizeZwork_host_sizseZ work_deviceZ work_hostŚ buffer_sizeZsyevdZ work_sizeZworkr r śc/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/cupy/linalg/_eigenvalue.pyŚ_syevd s„           ž   ü’’ ž’r)rcCs¢ddl}t |”t |”|jdkr5t |”\}}|j ”}t  |j dd…|”}t  |j |”}||fS|j dks=t j rK|j ||d”\}}||fSt||dƒS)a¢ Return the eigenvalues and eigenvectors of a complex Hermitian (conjugate symmetric) or a real symmetric matrix. Returns two objects, a 1-D array containing the eigenvalues of `a`, and a 2-D square array or matrix (depending on the input type) of the corresponding eigenvectors (in columns). Args: a (cupy.ndarray): A symmetric 2-D square matrix ``(M, M)`` or a batch of symmetric 2-D square matrices ``(..., M, M)``. UPLO (str): Select from ``'L'`` or ``'U'``. It specifies which part of ``a`` is used. ``'L'`` uses the lower triangular part of ``a``, and ``'U'`` uses the upper triangular part of ``a``. Returns: tuple of :class:`~cupy.ndarray`: Returns a tuple ``(w, v)``. ``w`` contains eigenvalues and ``v`` contains eigenvectors. ``v[:, i]`` is an eigenvector corresponding to an eigenvalue ``w[i]``. For batch input, ``v[k, :, i]`` is an eigenvector corresponding to an eigenvalue ``w[k, i]`` of ``a[k]``. .. warning:: This function calls one or more cuSOLVER routine(s) which may yield invalid results if input conditions are not met. To detect these invalid results, you can set the `linalg` configuration to a value that is not `ignore` in :func:`cupyx.errstate` or :func:`cupyx.seterr`. .. seealso:: :func:`numpy.linalg.eigh` rNé’’’’éT©Zcupyx.cusolverrZ_assert_stacked_2dZ_assert_stacked_squareŚsizerrrrrrŚndimrrrZsyevjr))rrŚcupyxŚ_r r!r$r"r r r(Śeigh^s     r1cCs„ddl}t |”t |”|jdkr*t |”\}}|j ”}t  |j dd…|”S|j dks2t j r:|j ||d”St||dƒdS)a- Compute the eigenvalues of a complex Hermitian or real symmetric matrix. Main difference from eigh: the eigenvectors are not computed. Args: a (cupy.ndarray): A symmetric 2-D square matrix ``(M, M)`` or a batch of symmetric 2-D square matrices ``(..., M, M)``. UPLO (str): Select from ``'L'`` or ``'U'``. It specifies which part of ``a`` is used. ``'L'`` uses the lower triangular part of ``a``, and ``'U'`` uses the upper triangular part of ``a``. Returns: cupy.ndarray: Returns eigenvalues as a vector ``w``. For batch input, ``w[k]`` is a vector of eigenvalues of matrix ``a[k]``. .. warning:: This function calls one or more cuSOLVER routine(s) which may yield invalid results if input conditions are not met. To detect these invalid results, you can set the `linalg` configuration to a value that is not `ignore` in :func:`cupyx.errstate` or :func:`cupyx.seterr`. .. seealso:: :func:`numpy.linalg.eigvalsh` rNr*r+Fr,)rrr/r0r r!r r r(Śeigvalsh“s    r2)F)r) rrZ cupy.cudarrZ cupy.linalgrZ cupy._corerr)r1r2r r r r(Śs      T5