o װi@sxUddlZddlZddlZddlmZdddDZiZeed<ddZ d d Z dd dZ ddZ ddZ dddZdS)N get_dtypecCsg|]}t|qSr.0xrrO/home/ubuntu/.local/lib/python3.10/site-packages/cupyx/scipy/sparse/_sputils.py sr )singledoublecsinglecdouble _upcast_memocCs t|tjSN) isinstancecupyndarrayrrrrisdense s rcCst|p t|o |jdkS)z8Is x either a scalar, an array scalar, or a 0-dim array?r)risscalarrndimrrrr isscalarlikesrrFcCsttjj}ttjj}tj}|dur||krtj}t|tjr%|f}|D]6}t|}t |j tjs]|rW|j dkr>q't |j tj rW|}|}||krW||krWq'tj}|Sq'|S)aBased on input (integer) arrays ``a``, determines a suitable index data type that can hold the data in the arrays. Args: arrays (tuple of array_like): Input arrays whose types/contents to check maxval (float, optional): Maximum value needed check_contents (bool, optional): Whether to check the values in the arrays and not just their types. Default: False (check only the types) Returns: dtype: Suitable index data type (int32 or int64) Nr)riinfoint32minmaxint64rrasarraycan_castdtypesize issubdtypeinteger)arraysmaxvalcheck_contentsint32minint32maxrarrminvalrrrget_index_dtypes0   r*cCst|dur6t|tr tdt|}tt|tjs$tdj|j dd|kr1dks8t dt ddSdS)NzlTuples are not accepted for the 'axis' parameter. Please pass in one of the following: {-2, -1, 0, 1, None}.z#axis must be an integer, not {name})namezaxis out of range) rtuple TypeErrortyperr!rr"format__name__ ValueError)axis axis_typerrr validateaxisFs r6cGsTt|}|dur |Stj|}tD]}t||r"|t|<|Sqtd|f)amReturns the nearest supported sparse dtype for the combination of one or more types. upcast(t0, t1, ..., tn) -> T where T is a supported dtype Examples: >>> upcast('int32') >>> upcast('int32','float32') >>> upcast('bool',float) Nz%no supported conversion for types: %r)rgetnumpy result_typesupported_dtypesrrr/)argstupcastrrrr=]s   r=c Cst|dkr tdt|dkr4zt|d}Wnty)t|df}Ynwtdd|D}n tdd|D}|dur\t|dkrKtd |ddksW|ddkr[td nst|}d d t |D}t|dkrt|}||krtd ||nMt|dkr|d}t|d|||dd}t ||\} } | dkrtdd|D} td || |d|| f||dd}ntdt|dkrtd|S)zCheck validity of the shaperz8function missing 1 required positional argument: 'shape'r-cs|]}t|VqdSroperatorindexrargrrr zcheck_shape..csr>rr?rBrrrrDrENz,shape must be a 2-tuple of positive integersz#'shape' elements cannot be negativecSsg|] \}}|dkr|qS)rr)rirrrrr szcheck_shape..z,cannot reshape array of size {} into shape{}css |] }|dkr dn|VqdS)rnewshapeNrrrrrrDsz&can only specify one unknown dimensionz$matrix shape must be two-dimensional) lenr/iterr@rAr.r3r8prod enumerater1divmod) r; current_shape shape_iter new_shape current_sizenegative_indexesnew_sizeskip specified unspecified remainder err_shaperrr check_shapezsN        "$ rY)rNFr)rr@r8cupy._core._dtyperr:rdict__annotations__rrr*r6r=rYrrrrs    0