`iddlZddlmZddlmZddddZdddd Zdddd Zdd Zdd Z d Z dZ dZ dZ dZdZdZdS)N)_core)fusionTKFblockingc 6tj|||||||S)aCreates an array on the current device. This function currently does not support the ``subok`` option. Args: obj: :class:`cupy.ndarray` object or any other object that can be passed to :func:`numpy.array`. dtype: Data type specifier. copy (bool): If ``False``, this function returns ``obj`` if possible. Otherwise this function always returns a new array. order ({'C', 'F', 'A', 'K'}): Row-major (C-style) or column-major (Fortran-style) order. When ``order`` is ``'A'``, it uses ``'F'`` if ``a`` is column-major and uses ``'C'`` otherwise. And when ``order`` is ``'K'``, it keeps strides as closely as possible. If ``obj`` is :class:`numpy.ndarray`, the function returns ``'C'`` or ``'F'`` order array. subok (bool): If ``True``, then sub-classes will be passed-through, otherwise the returned array will be forced to be a base-class array (default). ndmin (int): Minimum number of dimensions. Ones are inserted to the head of the shape if needed. blocking (bool): Default is ``False``, meaning if a H2D copy is needed it would run asynchronously on the current stream, and users are responsible for ensuring the stream order. For example, writing to the source ``obj`` without proper ordering while copying would result in a race condition. If set to ``True``, the copy is synchronous (with respect to the host). Returns: cupy.ndarray: An array on the current device. .. note:: This method currently does not support ``subok`` argument. .. note:: If ``obj`` is an `numpy.ndarray` instance that contains big-endian data, this function automatically swaps its byte order to little-endian, which is the NVIDIA and AMD GPU architecture's native use. .. seealso:: :func:`numpy.array` rarray)objdtypecopyordersubokndminrs l/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/cupy/_creation/from_data.pyr r s!\ ;sE4uh G GGc4tj||d||S)agConverts an object to array. This is equivalent to ``array(a, dtype, copy=False, order=order)``. Args: a: The source object. dtype: Data type specifier. It is inferred from the input by default. order ({'C', 'F', 'A', 'K'}): Whether to use row-major (C-style) or column-major (Fortran-style) memory representation. Defaults to ``'K'``. ``order`` is ignored for objects that are not :class:`cupy.ndarray`, but have the ``__cuda_array_interface__`` attribute. blocking (bool): Default is ``False``, meaning if a H2D copy is needed it would run asynchronously on the current stream, and users are responsible for ensuring the stream order. For example, writing to the source ``a`` without proper ordering while copying would result in a race condition. If set to ``True``, the copy is synchronous (with respect to the host). Returns: cupy.ndarray: An array on the current device. If ``a`` is already on the device, no copy is performed. .. note:: If ``a`` is an `numpy.ndarray` instance that contains big-endian data, this function automatically swaps its byte order to little-endian, which is the NVIDIA and AMD GPU architecture's native use. .. seealso:: :func:`numpy.asarray` Frr ar rrs rasarrayr8s @ ;q% A A AArc4tj||d||S)arConverts an object to array. This is currently equivalent to :func:`cupy.asarray`, since there is no subclass of :class:`cupy.ndarray` in CuPy. Note that the original :func:`numpy.asanyarray` returns the input array as is if it is an instance of a subtype of :class:`numpy.ndarray`. .. seealso:: :func:`cupy.asarray`, :func:`numpy.asanyarray` Frr rs r asanyarrayr[s ;q% A A AArc,tj||S)a)Returns a C-contiguous array. Args: a (cupy.ndarray): Source array. dtype: Data type specifier. Returns: cupy.ndarray: If no copy is required, it returns ``a``. Otherwise, it returns a copy of ``a``. .. seealso:: :func:`numpy.ascontiguousarray` )rascontiguousarray)rr s rrris  "1e , ,,rctjr4|dkrtdtjtj|S||S)aCreates a copy of a given array on the current device. This function allocates the new array on the current device. If the given array is allocated on the different device, then this function tries to copy the contents over the devices. Args: a (cupy.ndarray): The source array. order ({'C', 'F', 'A', 'K'}): Row-major (C-style) or column-major (Fortran-style) order. When ``order`` is ``'A'``, it uses ``'F'`` if ``a`` is column-major and uses ``'C'`` otherwise. And when ``order`` is ``'K'``, it keeps strides as closely as possible. Returns: cupy.ndarray: The copy of ``a`` on the current device. .. seealso:: :func:`numpy.copy`, :meth:`cupy.ndarray.copy` rz1cupy.copy does not support `order` in fusion yet.)r)r _is_fusingNotImplementedError _call_ufuncrelementwise_copyr )rrs rr r |sb,= C<<%CEE E!%"8!<<< 666  rc>ttj|i|S)zInterpret a buffer as a 1-dimensional array. .. note:: Uses NumPy's ``frombuffer`` and coerces the result to a CuPy array. .. seealso:: :func:`numpy.frombuffer` )rnumpy frombufferargskwargss rr"r"s" 5#T4V44 5 55rc>ttj|i|S)ayReads an array from a file. .. note:: Uses NumPy's ``fromfile`` and coerces the result to a CuPy array. .. note:: If you let NumPy's ``fromfile`` read the file in big-endian, CuPy automatically swaps its byte order to little-endian, which is the NVIDIA and AMD GPU architecture's native use. .. seealso:: :func:`numpy.fromfile` )rr!fromfiler#s rr'r's! 5>42622 3 33rc>ttj|i|S)zConstruct an array by executing a function over each coordinate. .. note:: Uses NumPy's ``fromfunction`` and coerces the result to a CuPy array. .. seealso:: :func:`numpy.fromfunction` )rr! fromfunctionr#s rr)r)s" 5%t6v66 7 77rc>ttj|i|S)zCreate a new 1-dimensional array from an iterable object. .. note:: Uses NumPy's ``fromiter`` and coerces the result to a CuPy array. .. seealso:: :func:`numpy.fromiter` )rr!fromiterr#s rr+r+s! 5>42622 3 33rc>ttj|i|S)zA new 1-D array initialized from text data in a string. .. note:: Uses NumPy's ``fromstring`` and coerces the result to a CuPy array. .. seealso:: :func:`numpy.fromstring` )rr! fromstringr#s rr-r-" 5#T4V44 5 55rc>ttj|i|S)zLoad data from a text file. .. note:: Uses NumPy's ``loadtxt`` and coerces the result to a CuPy array. .. seealso:: :func:`numpy.loadtxt` )rr!loadtxtr#s rr0r0s! 5=$1&11 2 22rc>ttj|i|S)zLoad data from text file, with missing values handled as specified. .. note:: Uses NumPy's ``genfromtxt`` and coerces the result to a CuPy array. .. seealso:: :func:`numpy.genfromtxt` )rr! genfromtxtr#s rr2r2r.r)NTrFr)NN)N)r)r!cupyr cupy._corerr rrrr r"r'r)r+r-r0r2rrr6sC .H.H.H.H.H.Hb B5 B B B B BF Be B B B B B----&D 6 6 6444"88844466633366666r