o Õ0 i”5ã@s†zddlZdZWn eydZYnwddlZddlmZddlmZddlZ ddlm Z ddlm Z Gdd „d e j ƒZ d d „ZdS) éNTF)Údriver)Úruntime)Ú_base)Ú _compressedc@sÈeZdZdZdZd/dd„Zdd„Zdd „Zd d „Zd d „Z dd„Z d0dd„Z dd„Z d1dd„Z d/dd„Zd1dd„Zdd„Zd2dd„Zdd „Zd!d"„Zd#d$„Zd%d&„Zd'd(„Zd)d*„Zd+d,„Zd-d.„ZdS)3Ú csc_matrixaâCompressed Sparse Column matrix. This can be instantiated in several ways. ``csc_matrix(D)`` ``D`` is a rank-2 :class:`cupy.ndarray`. ``csc_matrix(S)`` ``S`` is another sparse matrix. It is equivalent to ``S.tocsc()``. ``csc_matrix((M, N), [dtype])`` It constructs an empty matrix whose shape is ``(M, N)``. Default dtype is float64. ``csc_matrix((data, (row, col)))`` All ``data``, ``row`` and ``col`` are one-dimenaional :class:`cupy.ndarray`. ``csc_matrix((data, indices, indptr))`` All ``data``, ``indices`` and ``indptr`` are one-dimenaional :class:`cupy.ndarray`. Args: arg1: Arguments for the initializer. shape (tuple): Shape of a matrix. Its length must be two. dtype: Data type. It must be an argument of :class:`numpy.dtype`. copy (bool): If ``True``, copies of given arrays are always used. .. seealso:: :class:`scipy.sparse.csc_matrix` ÚcscNcCsHtstdƒ‚|j |¡}|j |¡}|j |¡}tjj|||f|j dS)a‰Returns a copy of the array on host memory. .. warning:: You need to install SciPy to use this method. Args: stream (cupy.cuda.Stream): CUDA stream object. If it is given, the copy runs asynchronously. Otherwise, the copy is synchronous. Returns: scipy.sparse.csc_matrix: Copy of the array on host memory. zscipy is not available)Úshape) Ú_scipy_availableÚ RuntimeErrorÚdataÚgetÚindicesÚindptrÚscipyÚsparserÚ_shape)ÚselfÚstreamr r r©rúc/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/cupyx/scipy/sparse/_csc.pyr 0s    ÿzcsc_matrix.getcCs@ddlm}| d¡r|j|dd}n| |¡}|j|j|jfS)Nr©ÚcusparseÚ denseToSparser)Úformat)ÚcupyxrÚcheck_availabilityrZ dense2cscr r r)rÚxrÚmrrrÚ_convert_denseFs   zcsc_matrix._convert_densecCs||fS©Nr)rrÚyrrrÚ_swapOszcsc_matrix._swapcCs¦ddlm}t |¡r| ¡| |j|¡Stjj  |¡ra| ¡| ¡|  d¡r9|  ¡}| ¡|  ||¡S|  d¡rLt jsL|j}|j||ddS|  d¡r_|  ¡}| ¡| ||¡St‚t|ƒr¼| ¡| ¡|  d¡r„t js„|j}|j}|j||dddS|  d¡rŸ|  ¡}|  ¡}| ¡| ¡| ||¡S|  d¡rº|  ¡}|  ¡}| ¡| ¡|  ||¡St‚tjj |¡rÉ||  ¡St |¡rQ|jdkrà| ¡| |j|¡S|jd kr| ¡|  d ¡rüt jrøt ¡d krü|j}n|  d ¡r t js |j}nt‚||jt |¡ddS|jd krM| ¡|  d¡r6t jr2t ¡d kr6|j}n |  d¡r@|j}nt‚||jt |¡ddStdƒ‚t S)NrrÚspgemmÚcsrgemmT)ÚtransaÚcsrgemm2)r$ZtransbéÚcsrmvi€ðúÚspmvéÚcsrmm2Úspmmzcould not interpret dimensions)!rrÚcupyZisscalarÚsum_duplicatesZ _with_datar rrZisspmatrix_csrrÚtocsrr"rÚis_hipÚTr#r%ÚAssertionErrorÚisspmatrix_cscZ isspmatrixrZisdenseÚndimrZget_build_versionr'r(Zasfortranarrayr*r+Ú ValueErrorÚNotImplemented)rÚotherrÚaÚbr'ZcsrmmrrrÚ__mul__Rs’              ÿý  ÿ ÿý zcsc_matrix.__mul__cCs0|j}| ¡|j}|j|_|j|_|j|_dS)zRemoves zero entories in place.N)r0Úeliminate_zerosr r r)rÚtÚcompressrrrr:°s  zcsc_matrix.eliminate_zeroscCs*ddlm}|js| |¡d|_dSdS)zŒSorts the indices of this matrix *in place*. .. warning:: Calling this function might synchronize the device. rrTN)rrZhas_sorted_indicesZcscsort)rrrrrÚ sort_indices¾s   þzcsc_matrix.sort_indicescCsÎddlm}|dur d}| ¡}|jdkrtj|j|j|dS| ¡}d|_ |  ¡|  d¡rOt j r7|jdkrO| |¡}|dkrB|S|dkrKt |¡Std ƒ‚|dkrZ| |j¡jS|dkrc| |¡Std ƒ‚) a’Returns a dense matrix representing the same value. Args: order ({'C', 'F', None}): Whether to store data in C (row-major) order or F (column-major) order. Default is C-order. out: Not supported. Returns: cupy.ndarray: Dense array representing the same matrix. .. seealso:: :meth:`scipy.sparse.csc_matrix.toarray` rrNÚC)rÚdtypeÚorderFÚ sparseToDenseÚFzorder not understood)rrÚupperZnnzr,Zzerosrr?ÚcopyÚhas_canonical_formatr-rrr/rAZascontiguousarrayr4Z csr2denser0Z csc2dense)rr@Úoutrrr rrrÚtoarrayËs0   ÿ    zcsc_matrix.toarraycCs`ddlm}| ¡| ¡j}| ¡| d¡r|j}n | d¡r%|j}nt‚||j|||ƒjS)NrrÚcsrgeam2Úcsrgeam) rrr-Útocscr0rrHrIÚNotImplementedError)rr6ÚalphaÚbetarrIrrrÚ _add_sparseøs    zcsc_matrix._add_sparseFcCs@ddlm}|r|j ¡}|j ¡}n|j}|j}| |||¡S)zóConverts the matrix to COOrdinate format. Args: copy (bool): If ``False``, it shares data arrays as much as possible. Returns: cupyx.scipy.sparse.coo_matrix: Converted matrix. rr)rrr rDr Zcsc2coo)rrDrr r rrrÚtocoos   zcsc_matrix.tocoocCs|r| ¡S|S)aConverts the matrix to Compressed Sparse Column format. Args: copy (bool): If ``False``, the method returns itself. Otherwise it makes a copy of the matrix. Returns: cupyx.scipy.sparse.csc_matrix: Converted matrix. ©rD)rrDrrrrJs zcsc_matrix.tocsccCs@ddlm}| d¡r|j}||ƒS| d¡r|j}||ƒSt‚)axConverts the matrix to Compressed Sparse Row format. Args: copy (bool): If ``False``, it shares data arrays as much as possible. Actually this option is ignored because all arrays in a matrix cannot be shared in csr to csc conversion. Returns: cupyx.scipy.sparse.csr_matrix: Converted matrix. rrÚcsc2csrÚ csc2csrEx2)rrrrQrRrK)rrDrrQrrrr..s  ûþzcsc_matrix.tocsrcCs| ¡S)zInverts the format. )r.)rrrrÚ_tocsxFszcsc_matrix._tocsxcCsP|durtdƒ‚|jd|jdf}tjjj|j|j|jf||d}|j |_ |S)aTReturns a transpose matrix. Args: axes: This option is not supported. copy (bool): If ``True``, a returned matrix shares no data. Otherwise, it shared data arrays as much as possible. Returns: cupyx.scipy.sparse.csr_matrix: `self` with the dimensions reversed. NzoSparse matrices do not support an 'axes' parameter because swapping dimensions is the only logical permutation.r&r)rrD) r4rrrrZ csr_matrixr r rrE)rZaxesrDrZtransrrrÚ transposeOs ÿÿzcsc_matrix.transposecCs|jt||dƒdd ¡S)zäReturns a copy of row i of the matrix, as a (1 x n) CSR matrix (row vector). Args: i (integer): Row Returns: cupyx.scipy.sparse.csc_matrix: Sparse matrix with single row r&TrP)Ú _minor_sliceÚslicer.©rÚirrrÚgetrowfs zcsc_matrix.getrowcCs|jt||dƒddS)zðReturns a copy of column i of the matrix, as a (m x 1) CSC matrix (column vector). Args: i (integer): Column Returns: cupyx.scipy.sparse.csc_matrix: Sparse matrix with single column r&TrP)Ú _major_slicerVrWrrrÚgetcolrs zcsc_matrix.getcolcCst||dƒ}| |¡ |¡S©Nr&)rVÚ_major_index_fancyrU©rÚrowÚcolrrrÚ_get_intXarray~ózcsc_matrix._get_intXarraycCs,t||dƒ}|jdv}| |¡j||dS)Nr&)r&NrP)rVÚsteprZrU)rr_r`rDrrrÚ_get_intXslice‚s zcsc_matrix._get_intXslicecCs"t||dƒ}| |¡j|ddS)Nr&TrP)rVrZrUr^rrrÚ_get_sliceXint‡szcsc_matrix._get_sliceXintcCó| |¡ |¡Sr)r]rUr^rrrÚ_get_sliceXarray‹ózcsc_matrix._get_sliceXarraycCst||dƒ}| |¡ |¡Sr\)rVrZÚ_minor_index_fancyr^rrrÚ_get_arrayXintŽrbzcsc_matrix._get_arrayXintcCrfr)rZrir^rrrÚ_get_arrayXslice’rhzcsc_matrix._get_arrayXslicer)NN)F)NF)Ú__name__Ú __module__Ú __qualname__Ú__doc__rr rr!r9r:r=rGrNrOrJr.rSrTrYr[rardrergrjrkrrrrrs0  ^ -       rcCs t|tƒS)zˆChecks if a given matrix is of CSC format. Returns: bool: Returns if ``x`` is :class:`cupyx.scipy.sparse.csc_matrix`. )Ú isinstancer)rrrrr2–s r2)Z scipy.sparserr Ú ImportErrorr,Zcupy_backends.cuda.apirrZcupyx.scipy.sparserrrZ_compressed_sparse_matrixrr2rrrrÚs" ÿ