
    JTh                     >    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)zWrapper 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       L/var/www/auris/envauris/lib/python3.13/site-packages/torch/testing/_utils.pywrapper_set_seedr      sj     
	"T$V$fell++0B0Bf0L KK!!# 
		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     
	 	 	,	,	.0J0J0L'')	::""$$"ZZ557N 1M	.+ [[$$002EHH4N4N4Pzz&&((

((8	* 5Q22 1M0L	.	. 5Q4P22U[[$$002EHH4N4N4Pzz&&((

((8	* 5Q4P4P222s   )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$   