o qiddlmZddlZddlmZddlmZmZmZmZm Z dde e zej zdefd Z deefd Zdd ede e zej zddfd Zd eeddfdZde ddfdZde ddfdZddZddZde fdZgdZdS))IterableN)Tensor) _lazy_call _lazy_initcurrent_device device_countis_initializedxpudevicereturncJtt|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 intindexrr default_generators get_state)r idxdefault_generators p/var/www/html/bestrading.cuttalo.com/services/ml-inference/venv/lib/python3.11/site-packages/torch/xpu/random.py get_rng_stater sLLL&#-f%% FC -eV,, ,C { 4S9  & & ( ((cPdttD}|S)zQReturn a list of ByteTensor representing the random number states of all devices.c,g|]}t|S)r).0is r z%get_rng_state_all.."s ???A}Q???r)ranger )resultss rget_rng_state_allr# s&??|~~)>)>???G Nr new_statectsVtj5tjdddn #1swxYwYt trtjn*t trtjddfd }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 r cj}|t}tjj|}|dSN)rrrr r set_state)rrr r$s rcbzset_rng_state..cb7sDl ; ""C!I8=##I.....rr N) r r_C_DisableFuncTorchclonecontiguous_formatrrr rr)r$r r*s`` r set_rng_stater0&s!   O X ' ' ) ) O O!e6MNNI O O O O O O O O O O O O O O O&#-f%% FC -eV,,///////rNNNNNs!AA #A  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) enumerater0)r1rstates rset_rng_state_allr5As< j))  5eQ  rseedcTtdfd }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`. r Nc|t}tjj|}|dSr()rrr r manual_seed)rrr6s rr*zmanual_seed..cbYs8!I8=%%d+++++rT)r6r+rrr6r*s` rr9r9KsI t99D,,,,,, rrcTtdfd }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. r NcttD].}tjj|}|/dSr()r!r rr rr9)rrr6s rr*zmanual_seed_all..cbksO|~~&& 0 0A % .cb}s5!I8=     rr+rr*s rr6r6ss'!!!! rNNNNNrc,dd}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. r Ncd}d}ttD][}tjj|}|s+||}d}F||\dS)NrFT)r!r rr rr6 initial_seedr9) random_seedseededrrs rr*zseed_all..cbs |~~&& ; ;A % > !--k:::: ; ;rr+rCrDs rr>r>s' ; ; ; ;rNNNNNrctt}tjj|}|S)zxReturn the current random seed of the current GPU. .. warning:: This function eagerly initializes XPU. )rrrr rrGrBs rrGrGs: LLL   C 4S9  ) ) + ++r) rr#r0r5r9r?r6r>rG)r r+)collections.abcrrrrrrr r rrr rlistr#r0r5r9r?r6r>rG__all__rrrrOs$$$$$$ RRRRRRRRRRRRRR))#)el2)v)))),4< VS3Y-ERV6 (6"2 t    cd,"#"$""""$$* ,c , , , ,   r