iddlmZddlZddlmZddlmZmZmZmZm Z dde e zejzdefdZ deefd Zdd ede e zejzddfd Zd eeddfd Zde ddfdZde ddfdZddZddZde fdZgdZy))IterableN)Tensor) _lazy_call _lazy_initcurrent_device device_countis_initializeddevicereturnc8tt|trtj|}n&t|t rtjd|}|j }| t}tjj|}|jS)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. xpu) r isinstancestrtorchr intindexrrdefault_generators get_state)r idxdefault_generators J/var/www/html/engine/venv/lib/python3.12/site-packages/torch/xpu/random.py get_rng_stater suL&#f% FC eV, ,,C { 44S9  & & ((cbttDcgc] }t|}}|Scc}w)zQReturn a list of ByteTensor representing the random number states of all devices.)ranger r)iresultss rget_rng_state_allr s,).|~)>?A}Q?G? N@s, new_statec~tsGtjj5j tj dddt trtjn&t trtjddfd }t|y#1swYhxYw)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_formatNrcj}| t}tjj|}|j yN)rrrrr set_state)rrr r s rcbzset_rng_state..cb7s<ll ; "C!II88=##I.rr N) r r_C_DisableFuncTorchclonecontiguous_formatrrr rr)r r r&s`` r set_rng_stater,&s   XX ' ' ) O!e6M6MNI O&#f% FC eV,/rN O Os !B33B< new_statescBt|D]\}}t||y)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-rstates rset_rng_state_allr1As& j) 5eQ rseedcBtdfd }t|dy)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`. Ncvt}tjj|}|j yr$)rrrr manual_seed)rrr2s rr&zmanual_seed..cbYs.!II88=%%d+rT)r2r'rrr2r&s` rr5r5Ks t9D, rrcBtdfd }t|dy)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. NcttD]0}tjj|}|j 2yr$)rr rrrr5)rrr2s rr&zmanual_seed_all..cbks<|~& 0A % < .cb}s+!II88= rr'rr&s rr2r2ss! rNrc"dd}t|y)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. Ncd}d}ttD]U}tjj|}|s#|j |j }d}E|j|Wy)NrFT)rr rrrr2 initial_seedr5) random_seedseededrrs rr&zseed_all..cbsg |~& ;A % < !--k: ;rr'r?r@s rr:r:s ;rNrctt}tjj|}|j S)zxReturn the current random seed of the current GPU. .. warning:: This function eagerly initializes XPU. )rrrrrrCr>s rrCrCs5 L  C 44S9  ) ) ++r) rrr,r1r5r;r2r:rC)rr')collections.abcrrrrrrr r rrr rlistrr,r1r5r;r2r:rC__all__rrrLs$ RR)#)ell2)v),4< VS3Y-ERV6 (6"2 t cd,"#"$"$$* ,c , r