o 0 i@sbddlZddlZddlmZddlmZddlmZddlmZddl m Z d dd Z d d Z dS) N)_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).0sdrrd/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/cupy/_manipulation/basic.py Ls zcopyto..) fillvaluez+could not broadcast input array from shape z into shape r)Z_where).typeboolintfloatcomplexnumpyemptyr copytorZ_FusionVarScalarrZ _ScalarProxycan_cast isinstanceZndarrayZisscalarsize ValueErroritem TypeErrorZ _is_fusingrZelementwise_copyZ _call_ufuncrZ _where_ufuncall itertools zip_longestreversedshapendimZsqueezetuplerange _can_memcpydataZcopy_from_asyncnbytesdevicerZ getDeviceZ setDeviceidcopy) dstsrcr whereZ src_is_scalarZsrc_typeZdst_arrrZ src_dtypeZ squeeze_ndimr+Z prev_devicerrrr s         rcCs@|jjo|jj}|jjo|jj}|s|o|j|jko|j|jkS)N)flags c_contiguous f_contiguousr r)r.r/r2r3rrrr(ss  r()rN) r!rZcupyrZ cupy._corerrZ cupy._sortingrZcupy_backends.cuda.apirrr(rrrrs       g