o X۷i@stddlmZddlZddlmZddlmZddlmZddZ e dfd d Z dd d Z e dfddZ dddZdS)) annotationsN)cuda)_new_like_order_and_strides)internalcCs:|dur |dur |j}|St|tr|f}|St|}|S)N)shape isinstanceinttuple)arr I/home/ubuntu/vllm_env/lib/python3.10/site-packages/cupyx/_pinned_array.py _update_shape s r CcCs@td|}t|t|j}t|}tj||||d}|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)rrrnbytesmemoutr r r empty_pinneds  rKc Csx|durtd|dur|j}t||}t||||dd\}}}t|t|j}t |}tj |||||d} | 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_rrrr r r empty_like_pinned.s     r!cCs t|||}tj|ddd|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)rrrrr r r zeros_pinnedZs r&cCs$t|||||}tj|ddd|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 rrrrrr r r zeros_like_pinnedrsr')NrNN) __future__rrcupyrcupy._creation.basicr cupy._corerr floatrr!r&r'r r r r s     ,