'}haddlmZmZmZddlZddlmZddlmZmZm Z m Z ddee e ejfdefd Zdeefd Z dd edee e ejfddfd Zd eeddfdZde ddfdZde ddfdZddZddZde fdZgdZy))IterableListUnionN)Tensor) _lazy_call _lazy_initcurrent_device device_countdevicereturnc8tt|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 intindexr rdefault_generators get_state)r idxdefault_generators O/var/www/html/test/engine/venv/lib/python3.12/site-packages/torch/xpu/random.py get_rng_statersuL&#f% FC eV, ,,C { 44S9  & & ((cng}ttD]}|jt||S)zQReturn a list of ByteTensor representing the random number states of all devices.)ranger appendr)resultsis rget_rng_state_allr"s3G <> ")}Q'() Nr new_statechtjj5|jtjdddt t rtjn&t trtjdfd}t|y#1swYgxYw)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)rr rrr set_state)rrr new_state_copys rcbzset_rng_state..cb7s<ll ; "C!II88=##N3r) r_C_DisableFuncTorchclonecontiguous_formatrrr rr )r#r r*r)s ` @r set_rng_stater/&s  # # %P"u7N7NOP&#f% FC eV,4rNPPs !B((B1 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/)r0r!states rset_rng_state_allr4As& j) 5eQ rseedc@tfd}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`. cvt}tjj|}|j yr')r rrr manual_seed)rrr5s rr*zmanual_seed..cbYs.!II88=%%d+rT)r5Nrr r5r*s` rr8r8Ks t9D, rrc@tfd}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. cttD]0}tjj|}|j 2yr')rr rrrr8)r!rr5s rr*zmanual_seed_all..cbks<|~& 0A % < as t9D0 rD!rc d}t|y)aoSet the seed for generating random numbers to a random number for the current GPU. It's safe to call this function if XPU 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`. crt}tjj|}|j yr')r rrrr5rrs rr*zseed..cb}s+!II88= rNr r*s rr5r5ss! rNrc d}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. cd}d}ttD]U}tjj|}|s#|j |j }d}E|j|Wy)NrFT)rr rrrr5 initial_seedr8) random_seedseededr!rs rr*zseed_all..cbsg |~& ;A % < !--k: ;rNrBrCs rr=r=s ;rNrctt}tjj|}|j S)zxReturn the current random seed of the current GPU. .. warning:: This function eagerly initializes XPU. )r r rrrrFrAs rrFrFs5 L  C 44S9  ) ) ++r) rr"r/r4r8r>r5r=rF)r)rN)typingrrrrrr r r r rrr rr"r/r4r8r>r5r=rF__all__rrrNs(( BB)%S%,, 67)F),4<@E$S#u||%;< 6 (6"2 t cd,"#"$"$$* ,c , r