Χg ddlmZmZmZddlZddlmZddlmZmZm Z m Z ddee e ej fdefd Zdeefd Z dd edee e ej fddfd Zd eeddfdZde ddfdZde ddfdZddZddZde fdZgdZdS))IterableListUnionN)Tensor) _lazy_call _lazy_initcurrent_device device_countxpudevicereturncJtt|trtj|}n*t|t rtjd|}|j}|t}tjj |}| S)aNReturn 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: ``'xpu'`` (i.e., ``torch.device('xpu')``, the current XPU device). .. warning:: This function eagerly initializes XPU. r ) r isinstancestrtorchr intindexr r default_generators get_state)r idxdefault_generators L/var/www/html/ai-engine/env/lib/python3.11/site-packages/torch/xpu/random.py get_rng_stater sLLL&#-f%% FC -eV,, ,C { 4S9  & & ( ((cg}ttD]$}|t|%|S)zQReturn a list of ByteTensor representing the random number states of all devices.)ranger appendr)resultsis rget_rng_state_allr! sDG <>> " "))}Q''(((( Nr new_statectj5|tjdddn #1swxYwYt t rtjn*t trtjdfd}t|dS)a,Set 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: ``'xpu'`` (i.e., ``torch.device('xpu')``, the current XPU device). ) memory_formatNr cj}|t}tjj|}|dSN)rr rr r set_state)rrr new_state_copys rcbzset_rng_state..cb9sDl ; ""C!I8=##N33333r) r_C_DisableFuncTorchclonecontiguous_formatrrr rr)r"r r)r(s ` @r set_rng_stater.(s   # # % %PP"u7NOOPPPPPPPPPPPPPPP&#-f%% FC -eV,,444444rNNNNNs!AAA new_statescPt|D]\}}t||dS)zSet the random number generator state of all devices. Args: new_states (Iterable of torch.ByteTensor): The desired state for each device. N) enumerater.)r/r states rset_rng_state_allr3Cs< j))  5eQ  rseedcRtfd}t|ddS)aSet the seed for generating random numbers for the current GPU. It's safe to call this function if XPU 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|t}tjj|}|dSr&)r rr r manual_seed)rrr4s rr)zmanual_seed..cb[s8!I8=%%d+++++rT)r4Nrrr4r)s` rr7r7MsD t99D,,,,, rrcRtfd}t|ddS)zSet the seed for generating random numbers on all GPUs. It's safe to call this function if XPU is not available; in that case, it is silently ignored. Args: seed (int): The desired seed. cttD].}tjj|}|/dSr&)rr rr rr7)r rr4s rr)zmanual_seed_all..cbmsO|~~&& 0 0A % .cbs5!I8=     rNrr)s rr4r4us"!!! rNNNNNrc*d}t|dS)zSet the seed for generating random numbers to a random number on all GPUs. It's safe to call this function if XPU is not available; in that case, it is silently ignored. cd}d}ttD][}tjj|}|s+||}d}F||\dS)NrFT)rr rr rr4 initial_seedr7) random_seedseededr rs rr)zseed_all..cbs |~~&& ; ;A % > !--k:::: ; ;rNrArBs rr<r<s" ; ; ;rNNNNNrctt}tjj|}|S)zxReturn the current random seed of the current GPU. .. warning:: This function eagerly initializes XPU. )r r rr rrEr@s rrErEs: LLL   C 4S9  ) ) + ++r) rr!r.r3r7r=r4r<rE)r )rN)typingrrrrrrr r r rrr rr!r.r3r7r=r4r<rE__all__rrrMs(((((((((( BBBBBBBBBBBB))%S%, 67)F)))),4<@E$S#u|%;< 6 (6"2 t    cd,"#"$""""$$* ,c , , , ,   r