`izUddlZddlZddlZddlmZddDZiZeed<dZ dZ dd Z d Z d Z dd ZdS)N get_dtypec,g|]}t|Sr.0xs o/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/cupyx/scipy/sparse/_sputils.py r s0@@@QIaLL@@@)singledoublecsinglecdouble _upcast_memoc6t|tjSN) isinstancecupyndarrayr s r isdenser s a & &&r c^tj|pt|o |jdkS)z8Is x either a scalar, an array scalar, or a 0-dim array?r)risscalarrndimrs r isscalarlikers( =   ; :qv{;r rFcptjtjj}tjtjj}tj}|||kr tj}t |tjr|f}|D]}tj|}tj |j tjsu|re|j dkrHtj |j tj r5|}|}||kr||krtj}n|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%arrminvals r get_index_dtyper0s"z$*%%)Hz$*%%)H JE  H  JE&$,''l3}SY 33  !8q==_SY ==! WWYYF WWYYF))f.@.@ JE E  Lr cb|t|trtdt|}t jt j|tjs(td|j d|cxkrdksntddSdS)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_types r validateaxisr=Fs  dE " " B@BB BJJ tz)44dlCC >A#V);V<<>> >da011 1'$ r ct|}||Stj|}tD]%}t j||r|t|<|cS&td|)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: )rgetnumpy result_typesupported_dtypesrr$r6)argstupcasts r rErE]s A}   %F  = # # !"L HHH  )ttE F FFr ct|dkrtdt|dkr[ t|d}td|D}nD#t$rt j|df}YnwxYwtd|D}|Kt|dkrt d|ddks |ddkrt d nDtj|}d t|D}t|dkr>tj|}||kr#t d ||nt|dkr|d}tj|d|||dzdz}t||\} } | dkrK|]}tj|VdSroperatorindexrargs r zcheck_shape..s,HHchnS11HHHHHHr c3>K|]}tj|VdSrrHrKs r rMzcheck_shape..s,>>#(.-->>>>>>r Nz,shape must be a 2-tuple of positive integersz#'shape' elements cannot be negativec$g|] \}}|dk |S)rr)rir s r r zcheck_shape..s!HHH$!Q!a%%A%%%r z,cannot reshape array of size {} into shape{}c3*K|]}|dkrdn|VdS)rnewshapeNrrs r rMzcheck_shape..s.!O!Oq1uu**!!O!O!O!O!O!Or z&can only specify one unknown dimensionz$matrix shape must be two-dimensional) lenr6iterr5rIrJr:r@prod enumerater8divmod) rC current_shape shape_iter new_shape current_sizenegative_indexesnew_sizeskip specified unspecified remainder err_shapes r check_shaperdzs 4yyA~~"## # Ta Id1gJHHZHHHHHII 4 4 4!Q003III 4 >>>>>>>  y>>Q  KLL L q\A  1!1!1BCC C"2z-00 HH)I*>*>HHH  A % %z),,H<'' "&&,f\9&E&EGGG(! " "a ' '#A&D 9QtV#4ya7I#IJJI%+L)%D%D "KA~~!!O!OY!O!O!OOO  "&&,f\9&E&EGGG!!D&)[N:YtAvww=OOIIEFF F 9~~?@@@ sA&&%B B)rNFr)rrIr@cupy._core._dtyperrBrdict__annotations__rrr0r=rErdrr r rhs  ''''''@@>@@@ d'''<<< ----`222.GGG://////r