`i5 ddlZdZn #e$rdZYnwxYwddlZddlmZddlmZddlZ ddlm Z ddlm Z Gdd e j Z d ZdS) NTF)driver)runtime)_base) _compressedceZdZdZdZddZdZdZdZdZ d Z dd Z d Z dd Z ddZddZdZddZdZdZdZdZdZdZdZdZdS) csc_matrixaCompressed 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` cscNctstd|j|}|j|}|j|}t j|||f|j S)aReturns 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 RuntimeErrordatagetindicesindptrscipysparser_shape)selfstreamrrrs k/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/cupyx/scipy/sparse/_csc.pyrzcsc_matrix.get0s  9788 8y}}V$$,""6**((|&& 7F #4;'88 8cddlm}|dr||d}n||}|j|j|jfS)Nrcusparse denseToSparser )format)cupyxrcheck_availabilityr dense2cscrrr)rxrms r_convert_densezcsc_matrix._convert_denseFsj""""""  & & 7 7 &&&q&77AA""1%%Avqy!(**rc ||fSN)rr!ys r_swapzcsc_matrix._swapOs 1v rc ddlm}tj|r1|||j|zStjj |r||| dr>| }|| ||S| dr+tjs|j}|||dS| dr>| }||||St$t'|rm||| dr3tjs'|j}|j}|||ddS| drf| }| }|||||S| drf| }| }||| ||St$tjj|r|| zSt+j|r|jdkr1|||j|zS|jd kr|| d r+tjrt1jd kr|j}n0| d rtjs|j}nt$||jtj|dS|jd kr|| dr+tjrt1jd kr|j}n$| dr|j}nt$||jtj|dSt?dt@S)NrrspgemmcsrgemmT)transacsrgemm2)r,transbcsrmvispmvcsrmm2spmmzcould not interpret dimensions)!rrcupyisscalarsum_duplicates _with_datarrrisspmatrix_csrrtocsrr*ris_hipTr+r-AssertionErrorisspmatrix_csc isspmatrixrisdensendimrget_build_versionr0r1asfortranarrayr3r4 ValueErrorNotImplemented)rotherrabr0csrmms r__mul__zcsc_matrix.__mul__Rs"""""" =  V "    ! ! !??49u#455 5 [  . .u 5 5S "    ! ! !  " " "**844 %JJLL  """q%000,,Y77 % %F''5'>>>,,Z88 %JJLL  """((E222$$ E " "B "    ! ! !  " " "**955 %gn %FG''1T$'GGG,,Z88 %JJLLKKMM  """  """((A...,,X66 %JJLLKKMM  """  """q!,,,$$ [  * *5 1 1, "%++--' ' ]5 ! !* "zQ##%%%ty5'8999q##%%%//88)$N)"355CC%NEE11&99) ')%MEE((uTVT%8%?%?MMMMq##%%%//99 )$N )"355CC%OEE0088)$MEE((uTVT%8%?%?MMMM !ABBB! !rc|j}||j}|j|_|j|_|j|_dS)zRemoves zero entories in place.N)r<eliminate_zerosrrr)rtcompresss rrLzcsc_matrix.eliminate_zerossB F 3M ' o rc\ddlm}|js||d|_dSdS)zSorts the indices of this matrix *in place*. .. warning:: Calling this function might synchronize the device. rrTN)rrhas_sorted_indicescscsort)rrs r sort_indiceszcsc_matrix.sort_indicessM #"""""& +   T " " "&*D # # # + +rcvddlm}|d}|}|jdkr!t j|j|j|S|}d|_ | | dr]tj r |jdkrF||}|dkr|S|dkrt j|St!d |dkr||jjS|dkr||St!d ) aReturns 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` rrNC)r dtypeorderF sparseToDenseFzorder not understood)rruppernnzr5zerosr rUcopyhas_canonical_formatr7rrr;rWascontiguousarrayrD csr2denser< csc2dense)rrVoutrr!r's rtoarrayzcsc_matrix.toarrays? #""""" =E  8q==:DJdjNNN N IIKK!&   ' ' 8 8 9  9+,5199&&q))A||#-a000 !7888||))!#..00#))!,,, !7888rcBddlm}||j}||dr|j}n$|dr|j}nt||j|||jS)Nrrcsrgeam2csrgeam) rrr7tocscr<rrdreNotImplementedError)rrFalphabetarres r _add_sparsezcsc_matrix._add_sparses""""""     & &z 2 2 &'GG  ( ( 3 3 &&GG% %wtvueT2244rFcddlm}|r3|j}|j}n|j}|j}||||S)zConverts 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)rrrr\rcsc2coo)rr\rrrs rtocoozcsc_matrix.tocoosj #"""""  #9>>##Dl''))GG9DlGdG444rc2|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. r\)rr\s rrfzcsc_matrix.tocscs  99;; Krcddlm}|dr|j}n$|dr|j}nt ||S)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. rrcsc2csr csc2csrEx2)rrrrqrrrg)rr\rrqs rr:zcsc_matrix.tocsr.sl #"""""  & &y 1 1 &&GG  ( ( 6 6 &)GG% %wt}}rc*|S)zInverts the format. )r:)rs r_tocsxzcsc_matrix._tocsxFszz||rc|td|jd|jdf}tjj|j|j|jf||}|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)r r\) rDr rrr csr_matrixrrrr])raxesr\r transs r transposezcsc_matrix.transposeOs  GHH H 1 tz!}, "-- Y dk 2%d.LL%)%>" rcv|t||dzdS)zReturns 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/Tro) _minor_sliceslicer:ris rgetrowzcsc_matrix.getrowfs5  q!a%t <<BBDDDrcR|t||dzdS)zReturns 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/Tro) _major_slicer|r}s rgetcolzcsc_matrix.getcolrs)  q!a%t <<>>CCC CCC>>>>>>>>>>>rrc,t|tS)zChecks if a given matrix is of CSC format. Returns: bool: Returns if ``x`` is :class:`cupyx.scipy.sparse.csc_matrix`. ) isinstancer)r!s rr>r>s a $ $$r) scipy.sparserr ImportErrorr5cupy_backends.cuda.apirrcupyx.scipy.sparserrr_compressed_sparse_matrixrr>r&rrrs ))))))******$$$$$$******D>D>D>D>D>6D>D>D>N %%%%%s