/usr/local/lib64/python3.6/site-packages/torch/cuda/__pycache__
Edit: /usr/local/lib64/python3.6/site-packages/torch/cuda/__pycache__/random.cpython-36.pyc (5520B)
3
Egg @ s d dl Z d dlmZmZmZmZ ddlmZmZm Z m
Z
ddlm Z ddd d
d d d
ddg Z d ee
ee jf e dddZee dddZd e ee
ee jf dddd Zee dddd
Ze
dddd Ze
dddd Zddd d
Zddd dZe
dd dZdS )! N)castIterableListUnion )
_lazy_init
_lazy_call device_countcurrent_device )Tensor
get_rng_stateget_rng_state_all
set_rng_stateset_rng_state_all manual_seedmanual_seed_allseedseed_all initial_seedcuda)devicereturnc C sZ t t| tr tj| } nt| tr2tjd| } | j}|dkrFt }tjj | }|j
S )aS Returns the random number generator state of the specified GPU as a ByteTensor.
Args:
device (torch.device or int, optional): The device to return the RNG state of.
Default: ``'cuda'`` (i.e., ``torch.device('cuda')``, the current CUDA device).
.. warning::
This function eagerly initializes CUDA.
r N) r
isinstancestrtorchr intindexr
r default_generatorsZ get_state)r idxdefault_generator r! =/usr/local/lib64/python3.6/site-packages/torch/cuda/random.pyr
s
)r c C s* g } x t t D ]}| jt| qW | S )zRReturns a list of ByteTensor representing the random number states of all devices.)ranger appendr
)resultsir! r! r" r " s ) new_stater r c sT | j tjdt tr$tj nt tr:tjd fdd}t| dS )a0 Sets the random number generator state of the specified GPU.
Args:
new_state (torch.ByteTensor): The desired state
device (torch.device or int, optional): The device to set the RNG state.
Default: ``'cuda'`` (i.e., ``torch.device('cuda')``, the current CUDA device).
)Z
memory_formatr c s6 t tj j} | d kr t } tjj| }|j d S )N)r r r r r
r r Z set_state)r r )r new_state_copyr! r" cb9 s
zset_rng_state.
.cbN)cloner Zcontiguous_formatr r r r r )r' r r) r! )r r( r" r + s
)
new_statesr c C s$ x t | D ]\}}t|| q
W dS )zSets the random number generator state of all devices.
Args:
new_states (Iterable of torch.ByteTensor): The desired state for each deviceN) enumerater )r+ r& stater! r! r" r C s )r r c s$ t fdd}t|dd dS )a Sets the seed for generating random numbers for the current GPU.
It's safe to call this function if CUDA is not available; in that
case, it is silently ignored.
Args:
seed (int): The desired seed.
.. warning::
If you are working with a multi-GPU model, this function is insufficient
to get determinism. To seed all GPUs, use :func:`manual_seed_all`.
c s t } tjj| }|j d S )N)r
r r r r )r r )r r! r" r) Z s zmanual_seed..cbT)r N)r r )r r) r! )r r" r L s c s$ t fdd}t|dd dS )zSets the seed for generating random numbers on all GPUs.
It's safe to call this function if CUDA is not available; in that
case, it is silently ignored.
Args:
seed (int): The desired seed.
c s. x(t t D ]} tjj| }|j q W d S )N)r# r r r r r )r& r )r r! r" r) l s zmanual_seed_all..cbT)r N)r r )r r) r! )r r" r b s c C s dd } t | dS )at Sets the seed for generating random numbers to a random number for the current GPU.
It's safe to call this function if CUDA is not available; in that
case, it is silently ignored.
.. warning::
If you are working with a multi-GPU model, this function will only initialize
the seed on one GPU. To initialize all GPUs, use :func:`seed_all`.
c S s t } tjj| }|j d S )N)r
r r r r )r r r! r! r" r) } s zseed..cbN)r )r) r! r! r" r t s c C s dd } t | dS )zSets the seed for generating random numbers to a random number on all GPUs.
It's safe to call this function if CUDA is not available; in that
case, it is silently ignored.
c S sP d} d}xBt t D ]4}tjj| }|s>|j |j } d}q|j| qW d S )Nr FT)r# r r r r r r r )Z random_seedZseededr& r r! r! r" r) s zseed_all..cbN)r )r) r! r! r" r s c C s t t } tjj| }|j S )zzReturns the current random seed of the current GPU.
.. warning::
This function eagerly initializes CUDA.
)r r
r r r r )r r r! r! r" r s )r )r )r typingr r r r r r r r
r __all__r r r r
r r r r r r r r r! r! r! r" s