ó
    "Eñi÷  ã                   ó>   • S SK r S SKrS r\ R                  S 5       rg)é    Nc                 ó:  • [        5          [        R                  " S5        U " U0 UD6n[        U[        R                  5      (       a8  UR
                  R                  S:X  a  [        R                  R                  5         UsSSS5        $ ! , (       d  f       g= f)z™Wrapper to set seed manually for some functions like dropout
See: https://github.com/pytorch/pytorch/pull/62315#issuecomment-896143189 for more details.
é*   ÚlazyN)	Úfreeze_rng_stateÚtorchÚmanual_seedÚ
isinstanceÚTensorÚdeviceÚtypeÚ_lazyÚ	mark_step)ÚopÚargsÚkwargsÚoutputs       ÚQ/home/mande/repo/quber/.venv/lib/python3.13/site-packages/torch/testing/_utils.pyÚwrapper_set_seedr      sj   € ô 
Õ	Ü×Ò˜"ÔÙ�TÐ$˜VÑ$ˆä�fœeŸl™l×+Ñ+°·±×0BÑ0BÀfÓ0Lô �K‰K×!Ñ!Ô#à÷ 
×	×	ús   ‹A7BÂ
Bc            	   #   ó
  #   • [         R                  R                  R                  5          [         R                  R                  5          [         R                  " 5       n [         R                  R                  5       (       a  [         R                  R                  5       nS S S 5        S S S 5         S v •  [         R                  R                  R                  5          [         R                  R                  5          [         R                  R                  5       (       a  [         R                  R                  W5        [         R                  " W 5        S S S 5        S S S 5        g ! , (       d  f       NÌ= f! , (       d  f       NÕ= f! , (       d  f       N9= f! , (       d  f       g = f! [         R                  R                  R                  5          [         R                  R                  5          [         R                  R                  5       (       a  [         R                  R                  W5        [         R                  " W 5        S S S 5        O! , (       d  f       O= fS S S 5        f ! , (       d  f       f = f= f7f)N)
r   ÚutilsÚ_mode_utilsÚno_dispatchÚ_CÚ_DisableFuncTorchÚget_rng_stateÚcudaÚis_availableÚset_rng_state)Ú	rng_stateÚcuda_rng_states     r   r   r      se  é € ô 
�‰×	 Ñ	 ×	,Ñ	,Õ	.´·±×0JÑ0JÕ0LÜ×'Ò'Ó)ˆ	Ü�:‰:×"Ñ"×$Ñ$Ü"ŸZ™Z×5Ñ5Ó7ˆN÷ 1M×	.ð+Ûô �[‰[×$Ñ$×0Ñ0Õ2´E·H±H×4NÑ4NÕ4PÜ�z‰z×&Ñ&×(Ñ(Ü—
‘
×(Ñ(¨Ô8Ü×Ò 	Ô*÷ 5Q×2Ð2÷ 1MÕ0Lú×	.Õ	.ú÷ 5QÕ4Pú×2Õ2ûŒU�[‰[×$Ñ$×0Ñ0Õ2´E·H±H×4NÑ4NÕ4PÜ�z‰z×&Ñ&×(Ñ(Ü—
‘
×(Ñ(¨Ô8Ü×Ò 	Ô*÷ 5Q×4PÖ4Pú×2×2Ö2ÿs¯   ‚)J«E8Á
AE'Â!E8Â)JÂ2F+ Â6(JÃFÃ=AF	ÅFÅ	JÅ'
E5	Å1E8Å8
FÆJÆ	
F	ÆFÆ
F(Æ$JÆ+)J ÇI/Ç3AI	É	I/É
I#ÉI/É&	J É/
I=É9J Ê J)Ú
contextlibr   r   Úcontextmanagerr   © ó    r   Ú<module>r%      s*   ðã ã òð  ×Ññ+ó ñ+r$   