`i`ddlZddlmZddlmZddlmZdZedfdZ d d Z edfd Z d d Z dS) N)cuda)_new_like_order_and_strides)internalcn| ||j}n(t|tr|f}nt|}|S)N)shape isinstanceinttuple)ars g/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/cupyx/_pinned_array.py _update_shaper s? } E3  e  LCctd|}tj|tj|jz}t j|}tj||||}|S)a"Returns a new, uninitialized NumPy array with the given shape and dtype. This is a convenience function which is just :func:`numpy.empty`, except that the underlying memory is pinned/pagelocked. Args: shape (int or tuple of ints): Dimensionalities of the array. dtype: Data type specifier. order ({'C', 'F'}): Row-major (C-style) or column-major (Fortran-style) order. Returns: numpy.ndarray: A new array with elements not initialized. .. seealso:: :func:`numpy.empty` N)dtypebufferorder) r rprodnumpyritemsizeralloc_pinned_memoryndarray)rrrnbytesmemouts r empty_pinnedrsa& $ & &E ]5 ! !EK$6$6$? ?F  "6 * *C -U3e D D DC JrKc@|td||j}t||}t||||d\}}}t j|t j|jz}tj |}t j |||||} | S)aReturns a new, uninitialized NumPy array with the same shape and dtype as those of the given array. This is a convenience function which is just :func:`numpy.empty_like`, except that the underlying memory is pinned/pagelocked. This function currently does not support ``subok`` option. Args: a (numpy.ndarray or cupy.ndarray): Base array. dtype: Data type specifier. The data type of ``a`` is used by default. order ({'C', 'F', 'A', or 'K'}): Overrides the memory layout of the result. ``'C'`` means C-order, ``'F'`` means F-order, ``'A'`` means ``'F'`` if ``a`` is Fortran contiguous, ``'C'`` otherwise. ``'K'`` means match the layout of ``a`` as closely as possible. subok: Not supported yet, must be None. shape (int or tuple of ints): Overrides the shape of the result. If ``order='K'`` and the number of dimensions is unchanged, will try to keep order, otherwise, ``order='C'`` is implied. Returns: numpy.ndarray: A new array with same shape and dtype of ``a`` with elements not initialized. .. seealso:: :func:`numpy.empty_like` Nzsubok is not supported yetF) get_memptr)rrstridesr) TypeErrorrr rrrrrrrr) r rrsubokrr _rrrs r empty_like_pinnedr$,s< 4555 } !U # #E3 5%5222E7A ]5 ! !EK$6$6$? ?F  "6 * *C -U3 'u 6 6 6C JrcVt|||}tj|dd|S)aReturns a new, zero-initialized NumPy array with the given shape and dtype. This is a convenience function which is just :func:`numpy.zeros`, except that the underlying memory is pinned/pagelocked. Args: shape (int or tuple of ints): Dimensionalities of the array. dtype: Data type specifier. order ({'C', 'F'}): Row-major (C-style) or column-major (Fortran-style) order. Returns: numpy.ndarray: An array filled with zeros. .. seealso:: :func:`numpy.zeros` runsafecasting)rrcopyto)rrrrs r zeros_pinnedr*Xs1& ueU + +C La**** JrcZt|||||}tj|dd|S)ajReturns a new, zero-initialized NumPy array with the same shape and dtype as those of the given array. This is a convenience function which is just :func:`numpy.zeros_like`, except that the underlying memory is pinned/pagelocked. This function currently does not support ``subok`` option. Args: a (numpy.ndarray or cupy.ndarray): Base array. dtype: Data type specifier. The dtype of ``a`` is used by default. order ({'C', 'F', 'A', or 'K'}): Overrides the memory layout of the result. ``'C'`` means C-order, ``'F'`` means F-order, ``'A'`` means ``'F'`` if ``a`` is Fortran contiguous, ``'C'`` otherwise. ``'K'`` means match the layout of ``a`` as closely as possible. subok: Not supported yet, must be None. shape (int or tuple of ints): Overrides the shape of the result. If ``order='K'`` and the number of dimensions is unchanged, will try to keep order, otherwise, ``order='C'`` is implied. Returns: numpy.ndarray: An array filled with zeros. .. seealso:: :func:`numpy.zeros_like` rr&r')r$rr))r rrr"rrs r zeros_like_pinnedr,ps56 AueUE : :C La**** Jr)NrNN) rcupyrcupy._creation.basicr cupy._corerr floatrr$r*r,rr r2s <<<<<<$34))))X$30r