o װi5@szddlZdZWn eydZYnwddlZddlmZddlmZddlZ ddlm Z ddlm Z Gdd d e j Z d d ZdS) NTF)driver)runtime)_base) _compressedc@seZdZdZdZd/ddZddZdd Zd d Zd d Z ddZ d0ddZ ddZ d1ddZ d/ddZd1ddZddZd2ddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.ZdS)3 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` cscNcCsHtstd|j|}|j|}|j|}tjj|||f|j dS)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)selfstreamr r rrK/home/ubuntu/.local/lib/python3.10/site-packages/cupyx/scipy/sparse/_csc.pyr 0s    zcsc_matrix.getcCs@ddlm}|dr|j|dd}n||}|j|j|jfS)Nrcusparse denseToSparser)format)cupyxrcheck_availabilityr dense2cscr r r)rxrmrrr_convert_denseFs   zcsc_matrix._convert_densecCs||fSNr)rryrrr_swapOszcsc_matrix._swapcCsddlm}t|r|||j|Stjj |ra||| dr9| }|| ||S| drLt jsL|j}|j||ddS| dr_| }||||Stt|r||| drt js|j}|j}|j||dddS| dr| }| }|||||S| dr| }| }||| ||Sttjj|r|| St|rQ|jdkr|||j|S|jd kr|| d rt jrtd kr|j}n| d r t js |j}nt||jt|ddS|jd krM|| dr6t jr2td kr6|j}n | dr@|j}nt||jt|ddStdt S)NrrspgemmcsrgemmT)transacsrgemm2)r%transbcsrmvispmvcsrmm2spmmzcould not interpret dimensions)!rrcupyisscalarsum_duplicates _with_datar rrisspmatrix_csrrtocsrr#ris_hipTr$r&AssertionErrorisspmatrix_csc isspmatrixrisdensendimrget_build_versionr)r*asfortranarrayr,r- ValueErrorNotImplemented)rotherrabr)csrmmrrr__mul__Rs                  zcsc_matrix.__mul__cCs0|j}||j}|j|_|j|_|j|_dS)zRemoves zero entories in place.N)r5eliminate_zerosr r r)rtcompressrrrrDs  zcsc_matrix.eliminate_zeroscCs*ddlm}|js||d|_dSdS)zSorts the indices of this matrix *in place*. .. warning:: Calling this function might synchronize the device. rrTN)rrhas_sorted_indicescscsort)rrrrr sort_indicess   zcsc_matrix.sort_indicescCsddlm}|dur d}|}|jdkrtj|j|j|dS|}d|_ | | drOt j r7|jdkrO||}|dkrB|S|dkrKt|Std |dkrZ||jjS|dkrc||Std ) 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)rdtypeorderF sparseToDenseFzorder not understood)rruppernnzr.zerosrrKcopyhas_canonical_formatr0rrr4rMascontiguousarrayr= csr2denser5 csc2dense)rrLoutrrr!rrrtoarrays0       zcsc_matrix.toarraycCs`ddlm}||j}||dr|j}n |dr%|j}nt||j|||jS)Nrrcsrgeam2csrgeam) rrr0tocscr5rrYrZNotImplementedError)rr?alphabetarrZrrr _add_sparses    zcsc_matrix._add_sparseFcCs@ddlm}|r|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)rrr rRr csc2coo)rrRrr r rrrtocoos   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. rR)rrRrrrr[s zcsc_matrix.tocsccCs@ddlm}|dr|j}||S|dr|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. rrcsc2csr csc2csrEx2)rrrrcrdr\)rrRrrcrrrr3.s  zcsc_matrix.tocsrcCs|S)zInverts the format. )r3)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)rrR) r=rrrr csr_matrixr r rrS)raxesrRrtransrrr transposeOs zcsc_matrix.transposecCs|jt||dddS)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(Trb) _minor_sliceslicer3rirrrgetrowfs zcsc_matrix.getrowcCs|jt||dddS)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(Trb) _major_slicerkrlrrrgetcolrs zcsc_matrix.getcolcCst||d}|||SNr()rk_major_index_fancyrjrrowcolrrr_get_intXarray~zcsc_matrix._get_intXarraycCs,t||d}|jdv}||j||dS)Nr()r(Nrb)rksteprorj)rrtrurRrrr_get_intXslices zcsc_matrix._get_intXslicecCs"t||d}||j|ddS)Nr(Trb)rkrorjrsrrr_get_sliceXintszcsc_matrix._get_sliceXintcC|||Sr )rrrjrsrrr_get_sliceXarrayzcsc_matrix._get_sliceXarraycCst||d}|||Srq)rkro_minor_index_fancyrsrrr_get_arrayXintrwzcsc_matrix._get_arrayXintcCr{r )ror~rsrrr_get_arrayXslicer}zcsc_matrix._get_arrayXslicer )NN)F)NF)__name__ __module__ __qualname____doc__rr rr"rCrDrIrXr_rar[r3rerirnrprvryrzr|rrrrrrrs0  ^ -       rcCs t|tS)zChecks if a given matrix is of CSC format. Returns: bool: Returns if ``x`` is :class:`cupyx.scipy.sparse.csc_matrix`. ) isinstancer)rrrrr7s r7) scipy.sparserr ImportErrorr.cupy_backends.cuda.apirrcupyx.scipy.sparserrr_compressed_sparse_matrixrr7rrrrs"