o ְi@sDddlZddlmZd ddZejfddZd ddZd d d ZdS) N)_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 K/home/ubuntu/.local/lib/python3.10/site-packages/cupy/_manipulation/kind.pyasarray_chkfinitesr cCs"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 asfarrays 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 r0s rc sddddddddd|s$ztj||dWSty#td|wz fdd|D}WntyGtd t|twd }|ddhkrTtd d|vr`d }| dn d|vrkd }| dd|v}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 eszrequire..z#Incorrect flag "{}" in requirementsAz%Cannot specify both "C" and "F" orderrrF)rrcopysubok) r asanyarray TypeErrorr formatKeyErrorsetkeyspopremovearray)r r requirementsrr$arrr r r requireAsH       r1)NN)N)rrr rrrr1r r r r s