o
    XÛ·ið  ã                   @  s,   d dl mZ d dlmZ dd„ Zdd„ ZdS )é    )Úannotations)Ú
_generatorc                 C  ó   t  ¡ }| | ¡S )z†Shuffles an array.

    Args:
        a (cupy.ndarray): The array to be shuffled.

    .. seealso:: :meth:`numpy.random.shuffle`

    )r   Úget_random_stateÚshuffle©ÚaÚrs© r
   úO/home/ubuntu/vllm_env/lib/python3.10/site-packages/cupy/random/_permutations.pyr      s   	
r   c                 C  r   )aa  Returns a permuted range or a permutation of an array.

    Args:
        a (int or cupy.ndarray): The range or the array to be shuffled.

    Returns:
        cupy.ndarray: If `a` is an integer, it is permutation range between 0
        and `a` - 1.
        Otherwise, it is a permutation of `a`.

    .. seealso:: :meth:`numpy.random.permutation`
    )r   r   Úpermutationr   r
   r
   r   r      s   
r   N)Ú
__future__r   Úcupy.randomr   r   r   r
   r
   r
   r   Ú<module>   s    