o W۷iT@sPddlmZddlZddlmZd ddZejfddZd dd Zd d d ZdS)) annotationsN)_corecCs*tj|||d}t|std|S)aConverts the given input to an array, and raises an error if the input contains NaNs or Infs. Args: a: array like. dtype: data type, optional order: {'C', 'F', 'A', 'K'}, optional Returns: cupy.ndarray: An array on the current device. .. note:: This function performs device synchronization. .. seealso:: :func:`numpy.asarray_chkfinite` )dtypeorderz#array must not contain Infs or NaNs)cupyasarrayisfiniteall ValueError)arrr M/home/ubuntu/vllm_env/lib/python3.10/site-packages/cupy/_manipulation/kind.pyasarray_chkfinitesrcCs"t|tjs tj}tj||dS)aConverts array elements to float type. Args: a (cupy.ndarray): Source array. dtype: str or dtype object, optional Returns: cupy.ndarray: The input array ``a`` as a float ndarray. .. seealso:: :func:`numpy.asfarray` r)r issubdtypeinexactfloat64rr rr r r asfarray s rcCs t||S)amReturn an array laid out in Fortran order in memory. Args: a (~cupy.ndarray): The input array. dtype (str or dtype object, optional): By default, the data-type is inferred from the input data. Returns: ~cupy.ndarray: The input `a` in Fortran, or column-major, order. .. seealso:: :func:`numpy.asfortranarray` )rasfortranarrayrr r r r2s rc sddddddddd|s$ztj||dWSty#td|wz fdd|D}WntyGtd t|twd }|ddhkrTtd d|vr`d }| dn d|vrkd }| dd|vrqdnd}z tj ||||dd}W|Stytd|w)aReturn an array which satisfies the requirements. Args: a (~cupy.ndarray): The input array. dtype (str or dtype object, optional): The required data-type. If None preserve the current dtype. requirements (str or list of str): The requirements can be any of the following * 'F_CONTIGUOUS' ('F', 'FORTRAN') - ensure a Fortran-contiguous array. * 'C_CONTIGUOUS' ('C', 'CONTIGUOUS') - ensure a C-contiguous array. * 'OWNDATA' ('O') - ensure an array that owns its own data. Returns: ~cupy.ndarray: The input array ``a`` with specified requirements and type if provided. .. seealso:: :func:`numpy.require` CFOWNDATA)r C_CONTIGUOUS CONTIGUOUSr F_CONTIGUOUSFORTRANOrrzIncorrect dtype "{}" providedcsh|]}|qSr )upper).0xpossible_flagsr r gszrequire..z#Incorrect flag "{}" in requirementsAz%Cannot specify both "C" and "F" orderrrTNF)rrcopysubok) r asanyarray TypeErrorr formatKeyErrorsetkeyspopremovearray)r r requirementsrr%arrr r!r requireCsH       r2)NN)N) __future__rrrrrrrr2r r r r s