o X۷i@sUddlmZddlZddlZddlZddlmZdddDZiZde d<d d Z d d Z dddZ ddZ ddZdddZdS)) annotationsN get_dtypecCsg|]}t|qSr.0xrrQ/home/ubuntu/vllm_env/lib/python3.10/site-packages/cupyx/scipy/sparse/_sputils.py sr )singledoublecsinglecdoubledict _upcast_memocCs t|tjSN) isinstancecupyndarrayrrrr isdenses 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_contentsint32minint32maxr!arrminvalrrr get_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#r!r$format__name__ ValueError)axis axis_typerrr validateaxisHs r8cGsTt|}|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_dtypesrr r1)argstupcastrrr r?_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@rrArDrrr rFrGNz,shape must be a 2-tuple of positive integersz#'shape' elements cannot be negativecSsg|] \}}|dkr|qS)rr)rirrrr r szcheck_shape..z,cannot reshape array of size {} into shape{}css |] }|dkr dn|VqdS)rnewshapeNrrrrr rFsz&can only specify one unknown dimensionz$matrix shape must be two-dimensional) lenr1iterrBrCr0r5r:prod enumerater3divmod) r= current_shape shape_iter new_shape current_sizenegative_indexesnew_sizeskip specified unspecified remainder err_shaperrr check_shape|sN        "$ r[)rNFr) __future__rrrBr:cupy._core._dtyperr<r__annotations__rrr,r8r?r[rrrr s   0