o W۷i@snddlmZddlZddlZddlmZddlmZddlmZddl m Z ddl m Z d d d Z d d ZdS)) annotationsN)_core)_fusion_interface)fusion)search)runtime same_kindc CsFd}t|}|ttttfvr#tjd|jd}tj|||dd}d}nG|t j t j fvr9|j}t ||j|}d}n1t|tjsDt|r_|jdkrMtd|j}t ||j|}|}d}n |j}t ||j|}|svtd||j|ft r|d urt||d St tj||||d S|std d tjt|jt|jdd Dstd |jd|j|j |j } | dkr|!t"t#| }|d urtj|||dd S|jdkrd S|rt||d St$||r|j%&|j%|j'd S|j(} t)*} zt)+| j,|j(| kr|-}t||Wt)+| d St)+| w)aCopies values from one array to another with broadcasting. This function can be called for arrays on different devices. In this case, casting, ``where``, and broadcasting is not supported, and an exception is raised if these are used. Args: dst (cupy.ndarray): Target array. src (cupy.ndarray): Source array. casting (str): Casting rule. See :func:`numpy.can_cast` for detail. where (cupy.ndarray of bool): If specified, this array acts as a mask, and an element is copied only if the corresponding element of ``where`` is True. .. seealso:: :func:`numpy.copyto` F)dtype)castingTz2non-scalar numpy.ndarray cannot be used for copytoz'Cannot cast %s to %s in %s casting modeNcSsg|] \}}||dfvqS)r r ).0sdr r N/home/ubuntu/vllm_env/lib/python3.10/site-packages/cupy/_manipulation/basic.py Ns zcopyto..) fillvaluez+could not broadcast input array from shape z into shape r)_where).typeboolintfloatcomplexnumpyemptyr copytor_FusionVarScalarr _ScalarProxycan_cast isinstancendarrayisscalarsize ValueErroritem TypeError _is_fusingrelementwise_copy _call_ufuncr _where_ufuncall itertools zip_longestreversedshapendimsqueezetuplerange _can_memcpydatacopy_from_asyncnbytesdevicer getDevice setDeviceidcopy) dstsrcr where src_is_scalarsrc_typedst_arrr src_dtype squeeze_ndimr7 prev_devicer r rrs         rcCs@|jjo|jj}|jjo|jj}|s|o|j|jko|j|jkS)N)flags c_contiguous f_contiguousr r")r<r=rFrGr r rr3us  r3)rN) __future__rr+rcupyr cupy._corerr cupy._sortingrcupy_backends.cuda.apirrr3r r r rs        g