o װi@sbddlZddlZddlmZddlmZddlmZddlmZdddZ dd d Z dd d Z dS)N)device)runtime)_util)_dtypeFcCsvddlm}ddlm}|dvrtdtj|dd\}}|j}|j} |j|d| d } |j \} } t | |} t d t j }tj}|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||jjW| |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|ddfS)Nr)cublas)cusolver)LUz UPLO argument must be 'L' or 'U'F)reject_float16F)ordercopyrfdFDz>Only float32, float64, complex64, and complex128 are supportedbfdD)r )+cupy_backends.cuda.libsrr ValueErrorrlinalg_common_typecharlowerastypeshapecupyemptynumpyint32rDevicecusolver_handleCUSOLVER_EIG_MODE_VECTORCUSOLVER_EIG_MODE_NOVECTORCUBLAS_FILL_MODE_LOWERCUBLAS_FILL_MODE_UPPERris_hip RuntimeErrorr to_cuda_dtype createParamsxsyevd_bufferSizedataptrxsyevdctypes destroyParamslinalg3_check_cusolver_dev_info_if_synchronization_allowedssyevd_bufferSizessyevddsyevd_bufferSizedsyevdcheevd_bufferSizecheevdzheevd_bufferSizezheevd)aUPLOwith_eigen_vector overwrite_arrdtypev_dtype real_dtypew_dtypevmldawdev_infohandlejobzuplotype_vtype_wparamswork_device_sizework_host_sizse work_device work_host buffer_sizesyevd work_sizeworkrrK/home/ubuntu/.local/lib/python3.10/site-packages/cupy/linalg/_eigenvalue.py_syevd s               rUrcCsddl}t|t||jdkr5t|\}}|j}t |j dd|}t |j |}||fS|j dks=t j rK|j||d\}}||fSt||dS)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` rNTcupyx.cusolverr_assert_stacked_2d_assert_stacked_squaresizerrrrrrndimrr%rsyevjrU)r9r:cupyx_r>r@rDrArrrTeigh^s     racCsddl}t|t||jdkr*t|\}}|j}t |j dd|S|j dks2t j r:|j||dSt||ddS)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` rNrVrWFrX)r9r:r_r`r>r@rrrTeigvalshs    rb)F)r) rr cupy.cudarr cupy.linalgr cupy._corerrUrarbrrrrTs      T5