"Ei*SSKJr SSKrSSKJr SSKJrJrJrJrJ r SS\ \ -\R-S\4Sjjr S\\4S jrSS \S\ \ -\R-SS4S jjrS \\SS4S jrS\ SS4SjrS\ SS4SjrSSjrSSjrS\ 4Sjr/SQrg))IterableN)Tensor) _lazy_call _lazy_initcurrent_device device_countis_initializeddevicereturncR[5 [U[5(a[R"U5nO,[U[ 5(a[R"SU5nUR nUc [5n[RRUnUR5$)a6Return 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 M/home/mande/repo/quber/.venv/lib/python3.13/site-packages/torch/xpu/random.py get_rng_stater s{L&#f% FC eV, ,,C { 44S9  & & ((ch[[55Vs/sHn[U5PM nnU$s snf)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^^[5(dD[RR5 TR [R S9mSSS5 [ T[5(a[R"T5mO,[ T[5(a[R"ST5mSUU4Sjjn[U5 g!,(df  N{=f)aSet 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_formatNrc>TRnUc [5n[RRUnUR T5 gN)rrrrr set_state)rrr r s rcbset_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 ' ' )!e6M6MNI*&#f% FC eV,//rN* )s C C new_statescB[U5Hup[X!5 M g)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_allr2As j)e*rseedc@^[T5mSU4Sjjn[USS9 g)alSet 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`. Ncv>[5n[RRUnUR T5 gr$)rrrr manual_seed)rrr3s rr&manual_seed..cbYs.!II88=%%d+rT)r3r(rrr3r&s` rr6r6Ks t9D, rrc@^[T5mSU4Sjjn[USS9 g)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. Nc>[[55H1n[RRUnUR T5 M3 gr$)rr rrrr6)rrr3s rr&manual_seed_all..cbks7|~&A % < as t9D0 rD!rc$SSjn[U5 g)a[Set 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`. Ncr[5n[RRUnUR 5 gr$)rrrrr3rrs rr&seed..cb}s+!II88= rr(rr&s rr3r3ss! rNrc$SSjn[U5 g)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. NcSnSn[[55H\n[RRUnU(d$UR 5 UR 5nSnMKURU5 M^ g)NrFT)rr rrrr3 initial_seedr6) random_seedseededrrs rr&seed_all..cbsb |~&A % < !--k:'rr(rCrDs rr=r=s ;rNrc[5 [5n[RRUnUR 5$)zlReturn the current random seed of the current GPU. .. warning:: This function eagerly initializes XPU. )rrrrrrGrAs rrGrGs5 L  C 44S9  ) ) ++r) rrr-r2r6r>r3r=rG)rr()collections.abcrrrrrrr r rrr rlistrr-r2r6r>r3r=rG__all__rrrQs$ RR)#)ell2)v),4< VS3Y-ERV6 (6"2 t cd,"#"$"$$* ,c , r